第一步:安装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字样说明成功

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐