windows安装tensorflow-gpu1.8.0+cuda9+cudnn7+anaconda记录
1安装anaconda
安装的时候要勾选配置环境变量,不然的话自己手动配置,最终的环境变量如下图
2安装显卡驱动
在nvidia网站安装自己型号的驱动
3安装cuda
安装的时候不要选择其中的显卡驱动,也不要选择nvidia experience
4安装cudnn
cudnn中的文件夹copy覆盖到cuda的目录中覆盖掉其中的同名文件
5测试cuda
没重启的时候输入nvcc -V 显示没有这个指令,但是在环境变量中多了这几个在系统变量中,
在path里面多了
重启之后在cmd中输入nvcc -V 显示正确的信息,因为需要重启使环境变量生效
6安装tensorflow
打开anaconda prompt,创建一个python3.6的环境,装tensorflow-gpu 1.8
pip install tensorflow-gpu==1.8.0
注意,一定要加上版本号,不然高版本的不支持cuda9.0
7测试
这里我是在jupyter notebook中测试的,关于如何notebook如何使用虚拟环境,见我的另一篇博客https://blog.****.net/qq_23996885/article/details/88691852
可以看到正在使用独立显卡GTX 860M