TensorFlow在Windows环境下的搭建

发表于

遵循:BY-SA

署名-相同方式共享 4.0协议

作者:谭东

时间:2017年5月28日

环境:Windows 7

首发地址:http://blog.csdn.net/jay100500/article/details/72792636

接下来,给大家讲一下在Windows下安装TensorFlow并运行一段简单程序的步骤及注意事项。

对于TensorFlow大家应该不是很陌生,其中AlphaGo就是用TensorFlow深度学习系统制作出来的。

给你们几个关键词:监督学习、半监督学习、神经网络、卷积神经网络、聚类、数据剪辑等。

DeepMind、Keras、Estimator、Torch。

当然,深度学习系统并不是只有TensorFlow,还有Caffe、Theano、keras、Torch、CNTK等。

看下TensorFlow特点:

(1)高度的灵活性:TensorFlow 不是一个严格的“神经网络”库。只要你可以将你的计算表示为一个数据流图,你就可以使用Tensorflow。

(2)真正的可移植性(Portability):Tensorflow 在CPU和GPU上运行,比如说可以运行在台式机、服务器、手机移动设备等等。

(3)将科研和产品联系在一起:过去如果要将科研中的机器学习想法用到产品中,需要大量的代码重写工作。那样的日子一去不复返了!

(4)自动求微分:基于梯度的机器学习算法会受益于Tensorflow自动求微分的能力。

(5)多语言支持:Tensorflow 有一个合理的c++使用界面,也有一个易用的Python使用界面来构建和执行你的graphs。

(6)性能最优化:比如说你又一个32个CPU内核、4个GPU显卡的工作站,想要将你工作站的计算潜能全发挥出来?由于Tensorflow 给予了线程、队列、异步操作等以最佳的支持,Tensorflow 让你可以将你手边硬件的计算潜能全部发挥出来。

任何人都可以用Tensorflow。学生、研究员、爱好者、极客、工程师、开发者、发明家、创业者等等都可以在Apache 2.0 开源协议下使用Tensorflow。

目前Google的TensorFlow是增加了Windows版本的支持,以前是只有Linux和MacOs版本。好了,那么我们就按照官方文档来安装吧。

官方文档地址:https://www.tensorflow.org/install/install_windows

有兴趣可以看下官方Windows版本安装英文原文文档,我也是看的这个。

1、首先在安装上有2个区分,如果你电脑支持GPU,那么你可以安装GPU版本,如果你的电脑不支持GPU,那么安装CPU版本。

先看看GPU版本需要多安装哪些。需要安装下面这2个驱动。

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源:

 

清华:https://pypi.tuna.tsinghua.edu.cn/simple

阿里云:http://mirrors.aliyun.com/pypi/simple/

中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/

华中理工大学:http://pypi.hustunique.com/

山东理工大学:http://pypi.sdutlinux.org/

豆瓣:http://pypi.douban.com/simple/

2.TensorFlow支持2种方式安装:

PIP方式,直接去Python官网去下载对应的Python Windows安装包就行了,要注意32位和64位版本的选择。

要注意,TensorFlow目前只能用Python 3.5.X版本的。所以不要下载错了版本。

下载链接:https://www.python.org/downloads/release/python-352/

64位和32位安装包。下载安装即可。

系统变量也加好了,安装好Python后,打开CMD,验证Python是否安装成功,输入:Python,出现Python版本号证明安装成功。

3.Python安装好后,我们就要进行下载安装TesorFlow需要的库了。

CPU版本命令输入:

pip3 install –upgrade tensorflow

GPU版本命令输入:

pip3 install –upgrade tensorflow-gpu

等待自动下载。

 

整个过程TensorFlow需要安装:numpy、six、wheel、appdirs、pyparsing、packaging、setuptools、protobuf、werkzeug、tensorflow。


4.安装完TensorFlow后,我们写一个例子进行测试下是否可以运行。在程序里找到:Python3.5或者CMD里输入Python即可进入输入编码模式。

给大家两个例子:

官方例子:


>>> import tensorflow as tf  
>>> hello = tf.constant('Hello, TensorFlow!')  
>>> sess = tf.Session()  
>>> print(sess.run(hello))

会输出:

Hello, TensorFlow!

另外一个计算的例子:


>>>import tensorflow as tf    
>>>sess = tf.Session()    
>>>a = tf.constant(10)    
>>>b = tf.constant(22)    
>>>print(sess.run(a + b))

会输出:

32

这样,我们的Windows下安装TensorFlow就完成了。

当然,TensorFlow不仅仅支持Python语法,还有如下这些扩展:

以上是”native” pip方式的安装,另一种方式Anaconda大家也可以试下。不过pip是官方推荐的,Anaconda不是官方支持的。


最后,通过国内阿里云镜像看下TensorFlow的相关吧。

接下来你就可以运用TensorFlow进行相关数据训练和深度学习了。

后续将进行TensorFlow的应用讲解…敬请期待..


文章评论
共收到 0 条评论