Я пытаюсь установить цепочку с поддержкой GPU и, следовательно, Cupy с поддержкой CUDA.
У меня установлена Cuda 10.1 и установить Cupy Wheel cupy-cuda101
.
Установка работает, нето проблеме сообщили.Однако, когда я импортирую цепочку, я получаю сообщение об ошибке ниже:
UserWarning: cuDNN is not enabled.
Please reinstall CuPy after you install cudnn
Насколько я понял, cudnn поставляется с этим колесом, и я действительно могу видеть dll в папке cupy cuda.Я попытался переустановить Cupy, как описано в ссылке в сообщении об ошибке, но это не помогло решить проблему.
Дополнительная информация:
chainer.cuda.available
True
chainer.cuda.cudnn_enabled
False
import cupy.cudnn
ModuleNotFoundError: No module named 'cupy.cudnn'
Так что я потерял все, что могу сделать, чтобырешить проблему.
РЕДАКТИРОВАТЬ:
Дополнительная информация: - Это новый и отдельный Python ENV в Anaconda - я фактически однажды уже удалил его снова и начал с нуля, но тот же вопрос