Windows上的TensorFlow:"无法打开CUDA库cudnn64_5.dll"

Question

Tensorflow刚刚发布了Windows支持.我安装了gpu版本和CUDA 8.0以及python 3.5.但是,在导入tensorflow后,我收到以下错误:

>>> import tensorflow
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:128] successfully opened CUDA library cublas64_80.dll locally
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:119] Couldn't open CUDA library cudnn64_5.dll
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\cuda\cuda_dnn.cc:3459] Unable to load cuDNN DSO
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:128] successfully opened CUDA library cufft64_80.dll locally
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:128] successfully opened CUDA library nvcuda.dll locally
I c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\dso_loader.cc:128] successfully opened CUDA library curand64_80.dll locally

有人可以帮忙吗？谢谢!

Answer 1

TL; DR:要将CuDNN与TensorFlow一起使用,该文件cudnn64_5.dll必须位于%PATH%环境变量中的目录中.请注意,cuDNN是从CUDA单独下载的,您必须下载cuDNN的5.1版.

首先,请注意cuDNN不与CUDA工具包的其余部分一起分发,因此您需要从NVIDIA网站单独下载.在Windows上,它作为ZIP存档分发,因此您必须将其解压缩并找到包含的目录cudnn64_5.dll.例如,如果将其解压缩到C:\tools\cuda,则DLL将处于C:\tools\cuda\bin\cudnn64_5.dll.最后,您可以通过在命令提示符下键入以下内容将其添加到路径中:

C:\> set PATH=%PATH%;C:\tools\cuda\bin
C:\> python
...
>>> import tensorflow as tf