Google Colab 上没有名为“cupy”的模块

Question

我正在开发 FastPhotoStyle 项目： https ://github.com/NVIDIA/FastPhotoStyle

我按照其教程的步骤进行操作： https://github.com/NVIDIA/FastPhotoStyle/blob/master/TUTORIAL.md

我正在 Google Colab 上运行示例 1，默认环境为

• CUDA 10.0
• Python 3.6
• 链接器5.4.0
• 铜吡咯5.4.0

这是我在 Colab Notebook 上的尝试：

https://colab.research.google.com/drive/1oBgdJgXCLlUQhpwPoG1Uom3OKTzHR4BF

运行后

!python3 demo.py --output_image_path results/example1.png

这是我收到的错误消息：

回溯（最近一次调用）：文件“demo.py”，第 9 行，位于

导入过程_风格化

文件“/content/drive/FastPhotoStyle/process_stylization.py”，第 14 行，位于

从 smooth_filter 导入 smooth_filter

文件“/content/drive/FastPhotoStyle/smooth_filter.py”，第 327 行，位于

从 cupy.cuda 导入函数

ModuleNotFoundError：没有名为“cupy”的模块

有人可以帮我吗？

Answer 1

您是否尝试过使用 python 完整路径运行演示？

!/usr/local/lib/python3.6//bin/python3 demo.py --output_image_path 

或者通过简单安装所有必需的库而无需安装 anaconda 即可运行

!pip install cupy

或在此处查看安装说明