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!'
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
大家在安装过程中有什么问题欢迎在评论区交流。