第一步:安装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

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐