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的版本就好