首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >Win10 Tensorflow-gpu 不完全安装手册

Win10 Tensorflow-gpu 不完全安装手册

作者头像
obaby
发布2023-02-24 14:56:35
发布2023-02-24 14:56:35
8200
举报
文章被收录于专栏:obaby@marsobaby@mars

网上随便搜一下就会发现关于Tensorflow-gpu的安装文章非常的多,但是写的都比较简略。并且官网的文档写的也比较的简略,并且google 官网上文档对于windows版本的也非常简略。

官网列出的硬件软件需求如下:

硬件要求 系统支持以下支持 GPU 的设备:

软件要求

必须在系统中安装以下 NVIDIA® 软件:

除此之外就没有更多的信息了,在官方的pip安装说明页面中可以看到windows版本的其实对于python是有要求的,官方支持的版本如下:

需要 Python 3.4、3.5 或 3.6

所以要安装tensorflow首先要找对python版本,建议用python 3.6 通过pip安装。这个可以看官方的安装说明:https://www.tensorflow.org/install/pip?hl=zh-cn

依赖库的安装需要注意一下,CUDA 需要用9.0版本 cuDNNsdk 需要7.2版本以上,而与CUDA 9.0对应的sdk版本为cudnn-9.0-windows10-x64-v7.4.2.24安装这连个版本即可。安装完成之后需要修改系统的环境变量,将安装路径加入到环境变量中:

确保安装的 NVIDIA 软件包与上面列出的版本一致。特别是,如果没有 cuDNN64_7.dll 文件,TensorFlow 将无法加载。要使用其他版本,请参阅在 Windows 下从源代码编译指南。 将 CUDA、CUPTI 和 cuDNN 安装目录添加到 %PATH% 环境变量中。例如,如果 CUDA 工具包安装到了 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 并且 cuDNN 安装到了 C:\tools\cuda,请更新 %PATH% 以匹配路径: SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin;%PATH% SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64;%PATH% SET PATH=C:\tools\cuda\bin;%PATH%

如果环境变量未生效,可以重启系统。

之后就可以运行测试脚本了:

效果如下:

代码语言:javascript
复制
import tensorflow as tf
tf.enable_eager_execution()
tf.add(1, 2).numpy()
hello = tf.constant('Hello, TensorFlow!')
print(hello.numpy())

☆文章版权声明☆

* 网站名称:obaby@mars

* 网址:https://h4ck.org.cn/

* 本文标题: 《Win10 Tensorflow-gpu 不完全安装手册》

* 本文链接:https://h4ck.org.cn/2019/01/win10-tensorflow-gpu-%e4%b8%8d%e5%ae%8c%e5%85%a8%e5%ae%89%e8%a3%85%e6%89%8b%e5%86%8c/

* 转载文章请标明文章来源,原文标题以及原文链接。请遵从 《署名-非商业性使用-相同方式共享 2.5 中国大陆 (CC BY-NC-SA 2.5 CN) 》许可协议。


分享文章:

相关文章:

  1. python3.7 安装alipay-sdk-python(PyCrypto)
  2. QQ音乐导出
  3. Django REST framework foreignkey 序列化
  4. .mht文件图片解析工具
  5. Python加载的文件哪里去了?(2)
  6. ncm2mp3
  7. m3u8 下载工具 V20.6.01 [OSX]
  8. 基于ffmpeg的m3u8下载[调整key替换逻辑,更新解析逻辑]
  9. 批处理 激活virtualenv 并且运行Python 命令
  10. CUDNN_STATUS_NOT_INITIALIZED
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2019年1月30日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 相关文章:
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档