Windows下安装TensorFlow教程

Win10 64位操作系统下tensorflow的安装

1,    安装Python和pip

TensorFlow是一个基于Python的机器学习框架，所以在安装TensorFlow之前，需要安装好Python。（已经安装好的可以跳过此步骤）Pip是python下的一个安装第三方库的工具，TensorFlow便是通过pip来安装的。

Python3.6  64位官方下载地址：

https://www.python.org/ftp/python/3.6.3/python-3.6.3-amd64.exe

32位官方下载地址：

https://www.python.org/ftp/python/3.6.3/python-3.6.3.exe

其他版本：

https://www.python.org/downloads/

下载安装完毕后，打开命令提示符，输入：python -V 确认是否安装成功

C:\Users\Dameng>python -V
Python 3.6.3

输入pip -V 确认pip是否可以正常使用

C:\Users\Dameng>pip -V
pip 9.0.1 from h:\python\python36\lib\site-packages (python 3.6)

如果pip无法使用，那么需要将python下的script文件夹加入Path环境变量中。

2. 安装TensorFlow

安装之前需要说明的是 TensorFlow 分为GPU版本和普通版本，如果你有一个支持CUDA的显卡，可以将机器学习中的计算压力交给强大的GPU去处理，能够大大提升TensorFlow的工作效率。（CUDA是NVIDIA推出的运算平台，所以A卡同学还是安心的下载普通版本吧）

鉴于大部分都是用的GeForce系列的显卡，就只给大家看下GeForce系列支持的显卡列表。别的架构显卡可以在这里查看是否支持：https://developer.nvidia.com/cuda-gpus

确定好自己安装GPU版本还是普通版本之后，就可以开始安装了。

使用终端/命令提示符

普通版本运行：pip install tensorflow

GPU版本运行：pip install tensorflow-gpu

等待安装完成即可。

普通版本的tensorflow到这里就安装完成了，可以写一段简单的tensorflow的代码验证一下：

import tensorflow as tf

HELLO = tf.constant("Hello TensorFlow!")
SESS = tf.Session()
print(SESS.run(HELLO))

用终端运行该文件之后便可输出

b'hello TensorFlow!'

如果是gpu版本的tensorflow运行，则会提示找不到cuda相关的库文件。接下来安装cuda和cudnn

3.    安装CUDA

CUDA目前NVIDIA官网已经发布了9.0版本，但是目前tensorflow还并不支持。目前安装的1.4.0版本的tensorflow支持CUDA8.0。（查看tensorflow的版本号请参考：http://blog.csdn.net/u011961856/article/details/76861052）

CUDA8.0的下载地址：https://developer.nvidia.com/cuda-80-download-archive

CUDA其他各个版本的下载地址：https://developer.nvidia.com/cuda-toolkit-archive

下载安装即可。

4.   安装CUDNN

CUDNN的版本需要与CUDA的版本对应：

Tensorflow 1.4.0版本对应的CUDNN版本为6.0，CUDA版本为8.0

CUDNN6.0下载地址：https://developer.nvidia.com/rdp/cudnn-download

（下载安装需要注册NVIDIA开发者）

下载完毕之后将压缩包中的bin目录里的东西解压到CUDA的bin目录里即可。

到此，TensorFlow的所有安装步骤均已完毕。

tips：

1. pip官方的下载速度可能会很慢，可以修改配置更改pip镜像地址，参考：https://www.biaodianfu.com/python-pip.html或http://blog.csdn.net/lsp1991/article/details/53411837

2. tensorflow运行中有时候会报一些warning，比如CPU支持AVX，AVX2但是tensorflow编译的时候未指定等等，觉得烦的可以将tensorflow的日志等级调高，就不会输出这类warning了，参考http://blog.csdn.net/otengyue/article/details/72823285

大家在安装过程中有什么问题欢迎在评论区交流。