第一步:安装cuda 、cudnn
cuda_10.0.130_411.31_win10.exe 安装时只勾选cuda安装,其他驱动不要动
cudnn-10.0-windows10-x64-v7.4.1.5
第二步:安装python pycharm
python-3.6.4-amd64.exe 加入环境变量
pycharm-professional-2019.1.3.exe 加入环境变量
第三步:安装tensorflow-gpu
直接通过pycharm安装,注意tensorflow版本和cuda的版本要对应
第四步:测试
import tensorflow as tf
#初始化一个Tensorflow的常量:Hello Google Tensorflow! 字符串,并命名为greeting作为一个计算模块
greeting = tf.constant('Hello Google Tensorflow!')
#启动一个会话
sess = tf.Session()
#使用会话执行greeting计算模块
result = sess.run(greeting)
print(result)
sess.close()
出现tensorflow字样说明成功