Windows10下安装GPU版本的Tensorflow

1、确保自己的显卡支持CUDA,安装CUDA8.0

2、下载cudnn6,解压后,修改环境变量,添加到路径,如下图:

Windows10下安装GPU版本的Tensorflow


3、安装Python3.5版本的anaconda3

4、命令行输入命令  pip install --upgrade tensorflow-gpu(没有pip的自行百度安装的方法)

5、个人安装之前遇到的问题:

安装的TensorFlow版本为1.3,之前看网上的教程使用的是cudnn5,import tensorflow的时候一直报错,

后来改为cudnn6之后就好了,个人认为是因为TensorFlow版本不一样的问题。