windows 安装tensorflow


2016年11月29日,TF官方宣布0.12版tensorflow支持原生windows操作系统,不在需要通过Docker进行安装。作为一个tf初学者,也是windows重度依赖用户,通过在墙里墙外各种搜索,终于找到了一种可行的安装方法。现予以总结,供同行参考。

博主机器配置:

[html]  view plain  copy
  1. OS:Window 7 64bit  
  2. CPU:Intel i7-2600K  
  3. 内存:8G  
  4. 显卡:Nvidia GeForce GTX 560  

(有人推荐使用 Windows PowerShell 代替 CMD,所以下面一、二、三、四步均在Power Shell下执行,“开始”->“附件”->“Windows Power Shell”->“Windows Power Shell”)

一、安装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”文件夹下,新建文本文件,添加内容:

[html]  view plain  copy
  1. [global]  
  2. index-url = http://mirrors.aliyun.com/pypi/simple/  
  3. [install]  
  4. trusted-host=mirrors.aliyun.com  

关于修改pip国内源可参考这里

四、安装TensorFlow

安装CPU版TensorFlow 0.12,Power Shell下输入:

[html]  view plain  copy
  1. 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下输入:

[html]  view plain  copy
  1. pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl  

官网查看最新版本下载方式

通过pip安装,会提示错误“Http error 404”。估计是无法联网导致的,可以将上述安装tf的命令https及后面的地址复制,并粘贴到浏览器地址栏中手动下载文件。

然后在Power Shell中输入下述命令实现本地安装:

[html]  view plain  copy
  1. pip install F:\DevResources\tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl  

出现错误:



不知什么原因,无法正常下载numpy 1.11.0。如法炮制,手动下载numpy(119MB)并安装。Numpy下载地址

我选择下载:numpy‑1.11.2+mkl‑cp35‑cp35m‑win_amd64.whl。其中1.11.2是nympy的版本,cp35表示支持python3.5,win_amd64就是支持windows64位操作系统。

下载完毕后先安装Numpy:

[html]  view plain  copy
  1. pip install F:\DevResources\numpy-1.11.2+mkl-cp35-cp35m-win_amd64.whl  

再安装TensorFlow(重复上面安装TensorFlow的步骤),安装成功。



五、测试TensorFlow

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

[html]  view plain  copy
  1. >>>import tensorflow as tf  
  2. >>>sess = tf.Session()  
  3. >>>a = tf.constant(10)  
  4. >>>b = tf.constant(22)  
  5. >>>print(sess.run(a + b))  
  6. 32  


正确输出结果32,安装完毕。


上述方法参考了很多网络资料,也有自己的尝试,如有不妥的地方还请大家批评指正!


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值