Как решить "cuDNN не включен" - PullRequest
       71

Как решить "cuDNN не включен"

1 голос
/ 01 декабря 2019

Я использую Cupy с Spyder3.3.6 и Python 3.7.5 на Windows-машине (Win10 Pro 64bit, i7-7700, 8GBMemory, GTX-1060-6GB).
Версия cupy, chainer, cuda и cuDNN6.0.0, 5.3.0, 10.1.243 и 7.6.4 соответственно.

Когда я импортирую Cupy, произошла эта ошибка:

C:\Users\(username)\Anaconda3\envs\gpgpu\lib\site-packages\chainer\backends\cuda.py:98: UserWarning: cuDNN is not enabled.
Please reinstall CuPy after you install cudnn
(see https://docs-cupy.chainer.org/en/stable/install.html#install-cudnn).
  'cuDNN is not enabled.\n'

Однако, попробуйте

import cupy

еще раз, ошибка не возникла, и Cupyработает хорошо. (т.е. напишите в коде «import cupy» 2 раза)
Однако Cupy действительно работает, мне это кажется странным.
Я уже пытался переустановить связанные пакеты, но проблема не решилась.

Как мне решить проблему?

1 Ответ

0 голосов
/ 02 декабря 2019

Прежде всего, кажется, что вы используете разные версии chainer / cupy. Мы рекомендуем цепочку и чашку, чтобы соответствовать номеру версии, поскольку мы разрабатываем их в тандеме.

Как вы установили CuPy? Мы поддерживаем предварительно созданные колеса для окон, которые включают версии cudnn и nccl, которые мы гарантируем для работы. Вы можете установить их с pip install cupy-cuda101.

...