如何解决“未启用cuDNN”

时间:2019-12-01 15:37:59

标签: python chainer cupy

我在Windows计算机(Win10 Pro 64位,i7-7700、8GB内存,GTX-1060-6GB)中使用带有Spyder3.3.6和Python 3.7.5的cupy。
cupy,chainer,cuda和cuDNN的版本分别为6.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工作正常。 (即在代码中写2次“ import cupy”)
但是cupy确实在起作用,这对我来说很奇怪。
我已经尝试过重新安装相关的软件包,但问题仍未解决。

我该如何解决问题?

1 个答案:

答案 0 :(得分:0)

首先,似乎您正在使用不同版本的链接器/ cupy。 我们建议串联使用chainer和cupy来匹配版本号。

您是如何安装CuPy的? 我们支持用于Windows的预制轮子,其中包括我们保证可以使用的cudnn和nccl版本。您可以使用pip install cupy-cuda101安装它们。