Tensorflow框架windows安装配置(二)

上次介绍了如何在windows配置非GPU版本的Tensorflow的安装,详情看这里。这次接着介绍GPU版本的如何安装。

1. CUDA、cuDNN安装

既然安装的是GPU版本,就需要安装CUDA和CUDNN,这个在笔者的文章中同样有说明,链接在这里。需要注意的是:cuDNN的版本选择是有规定的!cuDNN的版本选择是有规定的!cuDNN的版本选择是有规定的!往下看:

  • Tensorflow 1.0 到Tensorflow 1.2 版本使用的是cuDNN的5.1版本
  • Tensorflow 1.3 版本往后使用的是cuDNN的6.0版本。

cuDNN6.0版本可以在这里获取:(下载需要注册登录)https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v6/prod/8.0_20170307/cudnn-8.0-windows7-x64-v6.0-zip

2. Tensorflow-GPU安装

配置好CUDA和cuDNN之后,就是进行Tensorflow-GPU的安装了,使用如下命令:

pip install tensorflow-gpu==1.4.0

  • cudnn8.0的切记要输入1.4版本,否则更新版不支持,会出错!

3. 测试

在python中使用如下命令测试,效果图附上:

import tensorflow as tf # 引入 Tensorflow 库
gjm = tf.constant("Hello World ! I love TensorFlow ! ") #创建一个常量 Operation (操作)
sess = tf.Session() #启动一个 TensorFlow 的 Session(会话)
print (sess.run(gjm)) #运行 Graph (计算图) python3.x 专用语法
sess.close() #关闭 Session (会话)


成功安装tensorflow-gpu
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容