我在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/