用了一天的时间终于把GPU版本的Tensorflow安装成功,其中几个需要注意的事情;
1、电脑为Win7 64系统,显卡Nvida GT740M;
2、根据显卡到:https://developer.nvidia.com/cuda-downloads下在cuda,下载时候注意电脑的系统以及版本;
3、下载cudnn-8.0-windows7-x64-v6.0,注意cudnn要和cuda的版本对应,这是其中一个坑;
4,、安装好cuda之后,解压cudnn,将其中的三个文件放到cuda的对应文件夹内(cuda的安装目录),自己的是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
5、使用anaconda,目前使用的anaconda4.3版,支持python3.6,但是目前tensorflow-gpu版本只支持python3.5,通过conda命令建立python3.5的虚拟环境:conda create -n tensorflow python=3.5 anaconda,anaconda有个好处,建立此环境后,在anacond软件快捷键中会有下图Anaconda Prompt(tensorflow)的提示,使用时候,直接点击就可以进入该环境下;
总结一下:网上的关于win7系统安装tensorflow-gpu的教程很多,就不重复写了;
之前一直安装失败,出现下图的结果:
根据网上教程,主要需要注意:
1、安装cuda要和电脑系统对应,安装cudnn要和cuda的版本对应;安装cuda和cudnn之后一定要测试cuda是否安装成;如果测试过程报错,很可能是显卡驱动不是最新的,更新显卡驱动;(前提条件:cuda以及cudnn的版本对应都是正常的)
2、通过anaconda建立python3.5的虚拟环境。