2016年11月29日,TF官方宣布0.12版tensorflow支持原生windows操作系统,不在需要通过Docker进行安装。本文将讲述如何在windows下安装tensorflow。
(推荐使用 Windows PowerShell 代替 CMD,即,首先,点window+R,然后输入CMD,再输入Windows PowerShell。)
一、安装Python
1、通过Pip在Windows上安装python
TensorFlow在Windows上只支持64位Python3.5,可以通过Python 3.5 from python.org 或 Python 3.5 from Anaconda 下载并安装Python3.5.2(注意选择正确的操作系统)。
2、设置环境变量
上一步安装完毕后,在PowerShell中输入pip是找不到该命令的,因此将Python安装路径下“%安装路径%\Scripts”添加到Path下;再到PowerShell中输入pip看到若干命令提示,则代表python安装成功(Python安装包自带pip)。“开始”->“所有程序”,也可以找到Python终端。
二、安装Cuda和CuDNN(CPU版本TensorFlow请忽略本步骤)
TensorFlow分为CPU版和GPU版,如果你打算安装GPU版,请先安装如下两个驱动:
1、CUDA安装:https://developer.nvidia.com/cuda-downloads
2、CuDNN安装:https://developer.nvidia.com/cudnn(要注册Nvidia用户,并加入CuDNN开发组,填若干问卷就可以下载了)选择下载版本时要注意和Cuda版本匹配。解压后覆盖至CUDA的安装目录下
例如:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\
三、修改Pip国内源
为了使用国内镜像加速pip安装,需要如下修改:
WIndows 7 在“C:\Users\用户名\AppData\Local\pip”文件夹下,新建文本文件,添加内容:
[global]
index-url = http://mirrors.aliyun.com/pypi/simple/
[install]
trusted-host=mirrors.aliyun.com
关于修改pip国内源可参考这里
四、安装TensorFlow
安装CPU版TensorFlow 0.12,Power Shell下输入:
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-0.12.0rc0-cp35-cp35m-win_amd64.whl
安装GPU版TensorFlow 0.12,Power Shell下输入:
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl
下面是安装成功的截图:
五、测试TensorFlow
到“所有程序”下找到”Python 3.5 64bit”,出现命令窗口,输入测试代码:
>>>import tensorflow as tf
>>>sess = tf.Session()
>>>a = tf.constant(10)
>>>b = tf.constant(22)
>>>print(sess.run(a + b))
32
正确输出结果32,安装完毕。
注意: 我自己在安装的过程完成后,导入tensorflow时总是提示如下问题:
Error importing tensorflow. Unless you are using bazel.
you should not try to import tensorflow from its source directory;
please exit teh tensorflow source tree, and relaunch your python interpreter from there.
解决方法: 下载Windows 的 Microsoft Visual C++ 2015 redistributable update 3 64bit 安装即可
下载路径:https://www.microsoft.com/zh-cn/download/confirmation.aspx?id=53840
下面是程序跑通的截图
参考资料
【1】http://blog.csdn.net/hit1524468/article/details/66475606
【2】http://blog.csdn.net/include1224/article/details/53452824