噜噜噜,我又来讲述我的被坑之路了!!!安装了两天的TensorFlow-gpu,来回卸载cuda4次,来回重装TensorFlow-gpu数十次,我现在不来一吐苦水,我都对不起我自己,哼哼哼!!!
首先看安装步骤,直接pip install tensorflow-gpu,然后去官网下载了最新的cuda9.1+cudnn8.0,然后就报错了。。。暂时没有错误的截图了,反正报的错误就是:
ImportError: DLL load failed: 找不到指定的模块。During handling of the above exception, another exception occurred:找不到cudart64_90.dll,去官网下载cuda9.0。。。
开始以为是没有安装VS2015的锅,然而安装了VS2017 community之后,还是报错,呜呜呜呜。
看帖子说是要cuda8.0+cudnn6.0的组合,然后我一个个卸载了cudn后,重新安装了cuda8.0+cudnn6.0,本以为大功告成了,然而然而还是报错了。。。
正当我崩溃的同时,我看到了这个帖子,从此人生开始转运了!!!
点击打开链接 https://www.cnblogs.com/fanfzj/p/8521728.html
这个帖子是2018年3月的,比较新,帖子中安装的TensorFlow-gpu版本和我一样,都是1.7.0版的。然后,博主说道,安装的时候要注意TensorFlow-gpu的版本与cuda和cudnn的版本是否兼容!!!对的,我就是忽视了这个,在那儿来回重装了好几遍,呜呜呜。
重要的事情说三遍吧。注意TensorFlow-gpu的版本与cuda和cudnn的版本是否兼容!!!注意TensorFlow-gpu的版本与cuda和cudnn的版本是否兼容!!!注意TensorFlow-gpu的版本与cuda和cudnn的版本是否兼容!!!
因为我是TensorFlow-gpu1.7.0版的,所以需要安装cuda9.0+cudnn9.0。以前的帖子说的都是TensorFlow-gpu1.3.0,所以需要cuda8.0+cudnn6.0,哎呀,所以说呀,看帖子还是尽量看新的帖子吧。
成功安装了cuda9.0+cudnn9.0后,我的TensorFlow-gpu终于不报错了,上传一张截图,哈哈哈,撒花!!!
最后,希望我的踩坑大帖能够帮助到大家,嘻嘻嘻!!!