python - (Tensorflow-GPU) 导入tensorflow ImportError : Could not find 'cudnn64_7.dll'

Question

在anaconda下创建好tensorflow环境后，我安装了tensorflow-gpu。然后我试图导入 tensorflow 以验证它是否已正确安装，但出现此错误:

ImportError: Could not find 'cudnn64_7.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Note that installing cuDNN is a separate step from installing CUDA, and this DLL is often found in a different directory from the CUDA DLLs. You may install the necessary DLL by downloading cuDNN 7 from this URL: https://developer.nvidia.com/cudnn

设置是:

NVIDIA GTX 1080
CUDA 9.0
cuDNN 6.0
tensorflow-gpu 1.5

环境变量是:

CUDA_PAT: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
CUDA_PATH_V9_0: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

%Path% 变量是:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp
C:\Users\yshen\AppData\Local\cudnn-8.0-windows10-x64-v6.0\cuda\bin

很明显我安装的是cuDNN6.0，不知道为什么会报错“Could not find 'cudnn64_7.dll'”。为什么它会自动搜索 cudnn64_7.dll 而不是 cudnn64_6.dll？

Answer 1

此外，我在安装 TensorFlow 1.8 时遇到以下错误.我有 Anaconda 环境。

"ImportError: Could not find 'cudnn64_7.dll'"

但是在我安装之后Nvidia cuDNN v7.1.3 (2018 年 4 月 17 日)，对于 CUDA 9.0 ，一切开始工作。请注意，需要注册成为 Nvidia 开发人员才能下载安装包。

然后，只需按照页面中的说明进行操作即可:cudnn-install

对于 Windows:

3.3。在 Windows 上安装 cuDNN

以下步骤描述了如何构建依赖于 cuDNN 的程序。在以下部分中:

-您的 CUDA 目录路径称为 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

-您的 cuDNN 目录路径称为 <installpath>

1. 导航到您的 <installpath>包含 cuDNN 的目录.

2. 解压 cuDNN 包。 -cudnn-9.0-windows7-x64-v7.zip或 -cudnn-9.0-windows10-x64-v7.zip

3. 将以下文件复制到 CUDA 工具包目录中。

   • 复制<installpath>\cuda\bin\cudnn64_7.dll至 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin .
   • 复制<installpath>\cuda\ include\cudnn.h至 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include .
   • 复制<installpath>\cuda\lib\x64\cudnn.lib至 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64 .

4. 设置以下环境变量指向 cuDNN 所在的位置。访问 $(CUDA_PATH) 的值环境变量，执行以下步骤:

   • 从“开始”菜单打开命令提示符。
   • 键入运行并按 Enter。
   • 发出控制sysdm.cpl命令。
   • 选择窗口顶部的“高级”选项卡。
   • 点击窗口底部的环境变量。
   • 确保设置了以下值:变量名称:CUDA_PATH变量值:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

5. 在您的 Visual Studio 项目中包含 cudnn.lib。

   • 打开 Visual Studio 项目并 right-click在 project name 上.
   • 点击 Linker > Input > Additional Dependencies .
   • 添加cudnn.lib然后点击确定。