Windows上的TensorFlow:“无法打开CUDA库cudnn64_5.dll”
tensorflow
9
0

Tensorflow刚刚发布了Windows支持。我安装了gpu版本以及CUDA 8.0和python 3.5。但是,导入张量流后,出现以下错误:

>>> 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

有人可以帮忙吗?谢谢!

参考资料:
Stack Overflow
收藏
评论
共 2 个回答
高赞 时间 活跃

tensorflow-gpu之后尝试了pip3 install --upgrade tensorflow tensorflow-gpu ,它工作正常。

我认为这仅在尝试直接pip3 install --upgrade tensorflow-gpupip3 install --upgrade tensorflow-gpu问题。

收藏
评论

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
收藏
评论
新手导航
  • 社区规范
  • 提出问题
  • 进行投票
  • 个人资料
  • 优化问题
  • 回答问题

关于我们

常见问题

内容许可

联系我们

@2020 AskGo
京ICP备20001863号