Windows10下安装GPU版本的Tensorflow
1、确保自己的显卡支持CUDA,安装CUDA8.0
2、下载cudnn6,解压后,修改环境变量,添加到路径,如下图:
3、安装Python3.5版本的anaconda3
4、命令行输入命令 pip install --upgrade tensorflow-gpu(没有pip的自行百度安装的方法)
5、个人安装之前遇到的问题:
安装的TensorFlow版本为1.3,之前看网上的教程使用的是cudnn5,import tensorflow的时候一直报错,
后来改为cudnn6之后就好了,个人认为是因为TensorFlow版本不一样的问题。