Tensorflow+cuda+cudnn+window+Python之window下安装TensorFlow

kieojk 2017-03-07

展开全文

这篇文章主要讲TensorFlow在原生Windows下的安装教程。

安装环境：TensorFlow0.12+cuda8.0+cudnn v5.1+window7+python3.5

(1) 先安装Python3.5

从官网下载Python3.5，https://www./downloads/windows/，双击安装即可。接着将安装路径添加进环境变量中。具体步骤：计算机右键属性—高级系统设置—环境变量—系统变量—找到Path,把Python安装路径添加进去。

（2）安装Cuda和Cudnn（CPU版本的不需要安装，可直接跳过第二步）

首先要检查电脑的显卡型号是否支持GPU加速。

https://developer./cuda-gpus点击CUDA-EnabledGeForce Products。好像是computer capability大于3.0的才支持。我的显卡版本是GT720（低端显卡，凑合着用），支持GPU加速。

确定电脑显卡支持GPU加速后，开始下载Cuda: https://developer./cuda-downloads,（这里有个坑，如果浏览器下载一直下不了的话，建议用百度云离线下载）

下载完直接双击安装即可。

接下来下载Cudnn: https://developer./cudnn

Cudnn解压后将bin,include,lib三个文件夹里面的内容覆盖至Cuda安装目录下，默认路径为C:\Program Files\NVIDIA GPUComputing Toolkit\CUDA\v8.0（记住不是替换，是把Cudnn文件里的.dll文件添加到Cuda里面）

接下来，检验Cuda是否安装成功：

进入cuda路径，默认C:\ProgramData\NVIDIACorporation\CUDA Samples\v8.0\bin\win64\Release执行里面的deviceQuery.exe文件，如下图所示：

若出现上图结果，即说明你安装cuda成功。

（3）安装tensorflow

在安装TensorFlow之前，先修改镜像

国外镜像使用pip安装速度很慢，为了使用国内镜像加速pip安装，需要如下修改：

WIndows 7在“C:\Users\用户名\AppData\Local\pip”文件夹下，新建文本文件，添加内容：

1. [global]

2. index-url = http://mirrors.aliyun.com/pypi/simple/

3. [install]

4. trusted-host=mirrors.aliyun.com

ps:这里使用的是阿里云的国内镜像。

接着，在开始菜单打开windows PowerShell,输入py -3 –m pip install --upgrade https://storage./tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl

Ps:因为本人的电脑同时装了Python2.7和Python3.5，所以必须指明是使用哪一个版本的pip模块，若读者电脑只装了Python3.5，那么前面的py -3 –m语句可以忽略。

安装完成后打开Python 3.5 64bit 进行测试。

到“所有程序”下找到"Python 3.5 64bit"，出现命令窗口，输入测试代码：

1. >>>import tensorflow as tf

2. >>>sess = tf.Session()

3. >>>x = tf.constant(5)

4. >>>y = tf.constant(35)

5. >>>print(sess.run(x * y))

6. 175

测试成功，TensorFlow安装完成。

最后，检验一下cudnn是否配置成功：

Windows PowerShell 下输入python（因为我的电脑同时装了Python2.7和3.5，所以必须用py -3指定用哪个版本，若读者只装了Python3.5，那么直接输入python即可），再输入import tensorflow,若出现下图所示，不提示没有安装cudnn，则说明cudnn配置成功。