В Google Colab нет модуля с названием 'cupy'

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

и я следую шагам его учебника: https://github.com/NVIDIA/FastPhotoStyle/blob/master/TUTORIAL.md

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

  • CUDA 10.0
  • Python 3.6
  • Цепочка 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

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

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

import process_stylization

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

из smooth_filter импорт smooth_filter

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

из функции импорта cupy.cuda

ModuleNotFoundError: нет модуля с именем 'cupy'

Может кто-нибудь помочь мне с этим, пожалуйста?


python-3.x pytorch google-colaboratory cupy
person Amber Ji    schedule 12.06.2019    source источник
comment
ты сделал pip install cupy?   -  person Robert Crovella    schedule 12.06.2019
comment
Я сделал. ‹! CUDA_PATH = / opt / nvidia / cuda pip3 install cupy›. и он показывает Успешно установленный cupy-6.0.0   -  person Amber Ji    schedule 12.06.2019
comment
Может это связано. stackoverflow .com / questions / 49135065 /.   -  person corochann    schedule 13.06.2019
comment
@corochann Спасибо. Я уже видел этот пост, в котором в основном говорится, что у Colab есть встроенный CuPy. Я пробовал встроенный Cupy, но выскакивала та же ошибка ‹ModuleNotFoundError: Нет модуля с именем« cupy »›. Затем я установил CuPy вручную через ‹! CUDA_PATH = / opt / nvidia / cuda pip3 install cupy›, но возникла та же ошибка.   -  person Amber Ji    schedule 13.06.2019
comment
Как насчет попытки удалить существующую версию cupy pip uninstall cupy (введите это более 2 раз, чтобы убедиться, что все установленные версии Cupy, существующие в системе, удалены). И установить cupy с pip install cupy --no-cache-dir? Параметр --no-cache-dir важен для пересборки Cupy из исходников.   -  person corochann    schedule 13.06.2019
comment
@corochann Если я удалю какой-нибудь встроенный пакет в Colab (например, CuPy 5.4.0) сегодня, будет ли Colab автоматически восстанавливать его, когда я завтра перезапускаю ядро? Если нет, боюсь, что могу напортачить с CuPy, особенно когда дело касается переменных окружения.   -  person Amber Ji    schedule 13.06.2019

Ответы (1)


arrow_upward
1
arrow_downward

вы пробовали запустить демонстрацию с полным путем python?

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

или запуск без установки anaconda, просто установив все необходимые библиотеки

!pip install cupy
person Ioannis Nasios    schedule 13.06.2019
comment
Благодарность! Я работаю над этим. Пока что проблема в Анаконде. - person Amber Ji; 17.06.2019