如何解决“cuDNN 未启用”

Posted 2023-02-16

【中文标题】如何解决“cuDNN 未启用”

【英文标题】：How to solve "cuDNN is not enabled"

【发布时间】：2019-12-01 15:37:59

【问题描述】：

我在 Windows 机器（Win10 Pro 64bit、i7-7700、8GBMemory、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 运行良好。 （即在代码中写“import cupy”2 次） 然而cupy实际上在工作，这对我来说很奇怪。 我已经尝试过重新安装相关包，但问题没有解决。

我该如何解决这个问题？

【参考方案1】：

首先，您似乎正在使用不同版本的chainer/cupy。 我们建议chainer和cupy在我们串联开发时匹配版本号。

您是如何安装 CuPy 的？ 我们支持 Windows 的预构建***，其中包括我们保证可以工作的 cudnn 和 nccl 版本。您可以使用pip install cupy-cuda101 安装它们。

【讨论】：

感谢您的回答。我从 conda 安装了 CuPy。我可以通过 conda 而不是 pip 为 windows 安装 pre-vuilt ***吗？

Conda 构建不包括 cudnn 二进制文件，它们依赖于 conda。此外，conda 中的链接器版本已过时。目前，我们建议尽可能使用 pip 安装 chainer/cupy。如果 pip 不是一个选项，你总是可以尝试从源代码构建 cupy 和 chainer。