Windows10 TensorFlow “Couldn't open CUDA library cudnn64_5.dll”

環境爲:
Win10+TensorFlow1.0.1+GTX1060

在Windows10上配置TensorFlow with GPU,安裝完CUDA和cuDNN,運行程序 時發現:

“Couldn't open CUDA library cudnn64_5.dll”

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

原因主要是沒有找到cudnn64_5.dll,主要有兩種原因:
第一種,主要是沒有修改環境變量%PATH%,TensorFlow訓練加載cudnn時無法找到文件。
解決方案爲將cudnn文件拷貝到CUDA的默認目錄(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)下的bin,include,lib文件夾中,再將bin,include,lib路徑添加進環境變量中。
環境變量設置:


第二種,就是我遇見的問題,安裝時沒有注意cudnn版本要求,下載安裝了cudnnv6.0,安裝完後自己查找文件,只有cudnn64_6.dll,沒有cudnn64_5.dll。
解決方案爲將cudnnV6.0替換爲cudnnV5.1即可

發佈了7 篇原創文章 · 獲贊 21 · 訪問量 7萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章