В Google Colab нет модуля с именем cupy

Question

Я работаю над проектом FastPhotoStyle: https://github.com/NVIDIA/FastPhotoStyle

, и я следую инструкциям этого урока: https://github.com/NVIDIA/FastPhotoStyle/blob/master/TUTORIAL.md

Я запускаю Пример 1 в Google Colab, где по умолчаниюсреда

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

Вот как я пробовал на ноутбуке Colab:

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

После запуска

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

Вот сообщение об ошибке, которое я получил:

Traceback (последний вызов был последним): файл "demo.py", строка 9, в

import process_stylization

Файл "/content/drive/FastPhotoStyle/process_stylization.py", строка 14, в

из файла smooth_filter import smooth_filter

File "/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