声明:我使用的版本
-
tensorflow-gpu==1.4.0
-
CUDA (release 8.0, V8.0.60) 8.0版
-
cudnn(6.0)
但是最近在使用tensorflow时,突然报错
OSError: [WinError 126] 找不到指定的模块/Could not find ‘cudart64_6.dll’
原因:
在tensorflow调用NVIDIA时缺少了
cudart64_6.dll
,
cudart64_6.dll
是在cudnn6.0里的文件,这可能与我的软件自动升级有关,cudnn升级后造成和tensorflow1.4要求的版本不一致从而导致错误。
这里说明一下如何查看tensorflow-gpu对应的CUDA版本和cudnn版本的问题:(预先安装好Anaconda的前提下)
在路径
D:\Anaconda3\Lib\site-packages\tensorflow\python\platform
下可以看到
build_info.py
文件
版权声明:本文为qq_30622831原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。