Win10 Tensorflow-gpu 不完全安装手册

网上随便搜一下就会发现关于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%

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

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

 

效果如下:

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/
* 个性:https://oba.by/
* 本文标题: 《Win10 Tensorflow-gpu 不完全安装手册》
* 本文链接:https://image.h4ck.org.cn/2019/01/6330
* 短链接:https://oba.by/?p=6330
* 转载文章请标明文章来源,原文标题以及原文链接。请遵从 《署名-非商业性使用-相同方式共享 2.5 中国大陆 (CC BY-NC-SA 2.5 CN) 》许可协议。


You may also like

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注