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

时间:2019-06-12 13:57:34

标签: python-3.x pytorch google-colaboratory cupy

我正在研究FastPhotoStyle项目: https://github.com/NVIDIA/FastPhotoStyle

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

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

  • CUDA 10.0
  • Python 3.6
  • Chainer 5.4.0
  • CuPy 5.4.0

这是我在Colab Notebook上尝试的方式:

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

运行后

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

这是我收到的错误消息:

  

回溯(最近一次通话最近):文件“ demo.py”,第9行,位于

     

导入process_stylization

     

文件“ /content/drive/FastPhotoStyle/process_stylization.py”,第14行   在

     

从smooth_filter导入smooth_filter

     

文件

中的文件“ /content/drive/FastPhotoStyle/smooth_filter.py”,第327行      

来自cupy.cuda导入功能

     

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

有人可以帮我吗?

1 个答案:

答案 0 :(得分:0)

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

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

或者无需安装anaconda即可运行,只需安装所有必需的库即可

!pip install cupy