之前一直用的实验室的电脑,木有N卡,这次在自己的本本上整一下GPU版的tensorflow,配置不高,凑合着跑吧。。
CUDA版本为9.0,在官网下载了相应的cuDNN,版本为7.5.1,具体的安装过程就不说了,网上有很多教程,也不必再走一遍,这里遇到的最大问题就是tensorflow的版本,在anaconda中直接下载的最新gpu版1.13.1,运行简单的程序,显示错误:ImportError: DLL load failed: 找不到指定的模块。这一块花了不少时间,各种说辞,尝试了多种方案都不行,最后看到一个博主说这是由于tensorflow版本过高所致,当时一懵,难道高版本不支持低版本嘛,但这一块的确很乱,版本对应关系比较明确,我用的CUDA不是最新的,所以猜测tensorflow版本也应该是比较低的,于是使用以下命令将1.13.1降为1.5.0:
pip uninstall tensorflow-gpu==1.13.1
pip install tensorflow-gpu==1.5.0
之后在命令窗口测试成功: