我在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实际上正在工作,这对我来说很奇怪。
我已经尝试过重新安装相关的软件包,但问题仍未解决。

我该如何解决这个问题?

最佳答案

首先,看来您正在使用不同版本的链接器/ cupy。
当我们串联开发它们时,我们建议将chainer和cupy匹配版本号。

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

关于python - 如何解决“未启用cuDNN”,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/59127058/

10-14 03:38