正常的安装过程,请参考博客
WIN7配置tensorflow-gpu。
在这篇记录下我的测试过程。在使用anaconda 配置tensorflow-gpu时,发现它自动帮我们安装了CUDA8.0 vs cuDNN6.0。所以想尝试下不自己单独安装CUDA8.0 和cuDNN5.1,直接使用anaconda 配置tensorflow-gpu是否可行。
测试过程:
1)安装Anaconda3 5.1(安装完成后配置清华镜像或者中科大镜像)
2)安装一个编译环境 vc_redist.x64 (根据自身版本安装)
3)创建一个python3.5版本的环境,命令:conda create --name python35 python=3.5
4)在创建好的python35环境中安装tensorflow-gpu。这里我使用anaconda安装。
5)安装完成后,测试安装结果。依次输入,出现如下错误:
解决过程:
从tensorflow官方收集的问题中,发现有这个问题的解决办法。这个办法里面,也说是用CUDA8.0 和cuDNN5.1。所以按照这个方法,我安装了CUDA8.0 和cuDNN5.1。
安装完成后再进行测试,输入命令和相应代码后,会显示你的GPU信息和运算结果。测试结果:
Good luck !