如何将cudnn5.1升级到cudnn6

时间:2017-07-12 01:45:31

标签: tensorflow cudnn tensorflow-gpu

我已经按照Installing TensorFlow on Windows安装了tensorflow-gpu,当我导入tensorflow时似乎运行良好,但今天当我尝试新程序时,它显示了一些错误:

`

2017-07-12 09:33:31.301985: E c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\cuda\cuda_dnn.cc:352] Loaded runtime CuDNN library: 6021 (compatibility version 6000) but source was compiled with 5105 (compatibility version 5100).  If using a binary install, upgrade your CuDNN library to match.  If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.
2017-07-12 09:33:31.302851: F c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\kernels\conv_ops.cc:659] Check failed: stream->parent()->GetConvolveAlgorithms(&algorithms)

`

我知道这意味着我应该将我的cudnn5.1升级到cudnn6,然后我下载cudnn6并将文件复制到我的cuda安装目录。然后在目录cudn / v8.0 / bin中存在两个cudnn dll:cudnn64_5.dll和cudnn64_6.dll。

当我重新运行程序时,仍会出现相同的错误,当我删除cudnn64_5.dll时,我甚至无法运行我的程序,发生了一些新的错误:

`

builtins.ImportError: Traceback (most recent call last):
File "c:\Users\chenc\Anaconda3\envs\tensorflow-gpu\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py", line 18, in swig_import_helper
return importlib.import_module(mname)
File "c:\Users\chenc\Anaconda3\envs\tensorflow-gpu\lib\importlib\__init__.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "<frozen importlib._bootstrap>", line 986, in _gcd_import
File "<frozen importlib._bootstrap>", line 969, in _find_and_load
File "<frozen importlib._bootstrap>", line 958, in _find_and_load_unlocked
File "<frozen importlib._bootstrap>", line 666, in _load_unlocked
File "<frozen importlib._bootstrap>", line 577, in module_from_spec

`

我将我的cudnn64_6.dll重命名为cudnn64_5..dll,上面的错误消失但仍显示我的cudnn版本是5105.看起来我必须做其他事情来使cudnn升级工作,但我不会&#39 ;知道。我该怎么办才能解决这个错误?

1 个答案:

答案 0 :(得分:0)

我不确定如何解决您当前的问题,但我相信您不应该更新到cuDNN_6.0。 It is not currently supported with the latest version (1.2)并且不起作用。