Windows10安装TensorFlow-GPU版本

来源:互联网 发布:南柯奇谭 墨竹 知乎 编辑:程序博客网 时间:2024/05/27 01:03

Windows下安装TensorFlow的GPU版本首先准备如下:

1.Anaconda安装,Anaconda下载地址:Anaconda下载链接

2.安装tensorflow-gpu

3.安装CUDA8.0,下载地址:http://pan.baidu.com/s/1skZmSIp  提取密码:3wci

4.安装CUDNN6.0,下载地址:http://pan.baidu.com/s/1geFo6sV 提取密码:nj0c

需要注意的是电脑有GPU才能装这个版本,一般有NVIDIA显卡大部分是可以安装的

一、Anaconda下载安装

Anaconda下载地址:Anaconda下载链接,进去翻下面Anaconda3.0版本尽量装新一点的版本如下图

找到对应的版面win64位的点击下载。
我下载的版本为Anaconda3-4.3.1-Windows-x86_64

安装没什么特殊问题直接同意下一步,记住路径就行。

二、安装tensorflow-gpu

安装完Anaconda3之后利用pip安装tensorflow
这里需要注意的是不要在cmd下装,以管理员的身份进入cmd窗口,右键点击开始位置,点击Windows Powershell(管理员)(A)



在命令窗口输入:pip install tensorflow-gpu

由于我这安装成功了,未安装会下载安装。

三、安装CUDA8.0

CUDA8.0,下载地址:http://pan.baidu.com/s/1skZmSIp  提取密码:3wci
下载解压点击这个另一个文件是CUDNN5不用管。直接装CUDA8.0
和正常软件一样没什么特殊状况,只是在安装的时候屏幕会闪,不用担心正常等待安装。

四、安装CUDNN6.0

下载地址:http://pan.baidu.com/s/1geFo6sV 提取密码:nj0c
下载解压,放到任意盘
比如我放在C盘下面

然后把cuda/bin添加到环境变量
点击控制面板--系统和安全--系统--高级系统设置--环境变量

进去后找到系统变量中的Path


新建输入C:\cuda\bin
根据自己放置cuda路径不同修改即可。

五、测试tensorflow-GPU能否使用

测试代码
import tensorflow as tfhello = tf.constant('Hello,Tensorflow!')sess = tf.Session()print(sess.run(hello))
输出结果如下已经成功在GPU上跑。