1.安装CUDA9.0

目前tensorflow支持到9.0所以就用9.0了,官网下载安装包安装。

cmd中输入nvcc --version 命令之后显示版本号就完美安装了

2.CUDNN库

把CUDNN7.0 for CUDA9.0的库安装相应位置复制到CUDA的相应位置

CUDA的默认路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

3.安装TensorFlow-GPU版本

打开anaconda,新建虚拟环境之后

pip install --upgrade --ignore-installed tensorflow-gpu

等安装完了就好

4.测试是否启用了

经典的Hello world

import tensorflow as tf
hello = tf.constant("Hello!TensorFlow")
sess = tf.Session()
print(sess.run(hello))

如果是GPU版本的话,会显示显卡的一些信息,最终显示Hello!TensorFlow视为环境配置完成

5.Pycharm配置TensorFlow-GPU的编译

手动添加位于Anaconda的python版本,具体位置在Anaconda Navigator里面可以看到,记得python选3.5的版本就好