TensorFlow在Windows環(huán)境下的搭建

2018-09-01 21:25 更新

遵循:BY-SA署名-相同方式共享 4.0協(xié)議

作者:譚東

時間:2017年5月28日

環(huán)境:Windows 7

接下來,給大家講一下在Windows下安裝TensorFlow并運(yùn)行一段簡單程序的步驟及注意事項(xiàng)。

對于TensorFlow大家應(yīng)該不是很陌生,其中AlphaGo就是用TensorFlow深度學(xué)習(xí)系統(tǒng)制作出來的。

給你們幾個關(guān)鍵詞:監(jiān)督學(xué)習(xí)、半監(jiān)督學(xué)習(xí)、神經(jīng)網(wǎng)絡(luò)、卷積神經(jīng)網(wǎng)絡(luò)、聚類、數(shù)據(jù)剪輯等。

DeepMind、Keras、Estimator、Torch。

當(dāng)然,深度學(xué)習(xí)系統(tǒng)并不是只有TensorFlow,還有Caffe、Theano、keras、Torch、CNTK等。

看下TensorFlow特點(diǎn):

(1)高度的靈活性:TensorFlow 不是一個嚴(yán)格的“神經(jīng)網(wǎng)絡(luò)”庫。只要你可以將你的計(jì)算表示為一個數(shù)據(jù)流圖,你就可以使用Tensorflow。

(2)真正的可移植性(Portability):Tensorflow 在CPU和GPU上運(yùn)行,比如說可以運(yùn)行在臺式機(jī)、服務(wù)器、手機(jī)移動設(shè)備等等。

(3)將科研和產(chǎn)品聯(lián)系在一起:過去如果要將科研中的機(jī)器學(xué)習(xí)想法用到產(chǎn)品中,需要大量的代碼重寫工作。那樣的日子一去不復(fù)返了!

(4)自動求微分:基于梯度的機(jī)器學(xué)習(xí)算法會受益于Tensorflow自動求微分的能力。

(5)多語言支持:Tensorflow 有一個合理的c++使用界面,也有一個易用的Python使用界面來構(gòu)建和執(zhí)行你的graphs。

(6)性能最優(yōu)化:比如說你又一個32個CPU內(nèi)核、4個GPU顯卡的工作站,想要將你工作站的計(jì)算潛能全發(fā)揮出來?由于Tensorflow 給予了線程、隊(duì)列、異步操作等以最佳的支持,Tensorflow 讓你可以將你手邊硬件的計(jì)算潛能全部發(fā)揮出來。

任何人都可以用Tensorflow。學(xué)生、研究員、愛好者、極客、工程師、開發(fā)者、發(fā)明家、創(chuàng)業(yè)者等等都可以在Apache 2.0 開源協(xié)議下使用Tensorflow。

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

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

有興趣可以看下官方Windows版本安裝英文原文文檔,我也是看的這個。

1、首先在安裝上有2個區(qū)分,如果你電腦支持GPU,那么你可以安裝GPU版本,如果你的電腦不支持GPU,那么安裝CPU版本。

先看看GPU版本需要多安裝哪些。需要安裝下面這2個驅(qū)動。

1)CUDA   https://developer.nvidia.com/cuda-downloads

2)cuDNN  https://developer.nvidia.com/cudnn(要注冊Nvidia用戶,并加入CuDNN開發(fā)組,填若干問卷就可以下載了)選擇下載版本時要注意和Cuda版本匹配。解壓后覆蓋至CUDA的安裝目錄下

例如:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\

國內(nèi)PIP源:

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

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

中國科技大學(xué) https://pypi.mirrors.ustc.edu.cn/simple/

華中理工大學(xué):http://pypi.hustunique.com/

山東理工大學(xué):http://pypi.sdutlinux.org/ 

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

2.TensorFlow支持2種方式安裝:

PIP方式,直接去python官網(wǎng)去下載對應(yīng)的Python Windows安裝包就行了,要注意32位和64位版本的選擇。

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

下載鏈接:https://www.python.org/downloads/release/python-352/

64位和32位安裝包。下載安裝即可。

系統(tǒng)變量也加好了,安裝好Python后,打開CMD,驗(yàn)證Python是否安裝成功,輸入:Python,出現(xiàn)Python版本號證明安裝成功。

3.Python安裝好后,我們就要進(jìn)行下載安裝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后,我們寫一個例子進(jìn)行測試下是否可以運(yùn)行。在程序里找到:Python3.5或者CMD里輸入Python即可進(jìn)入輸入編碼模式。

給大家兩個例子:

官方例子:

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

會輸出:

Hello, TensorFlow!

另外一個計(jì)算的例子:

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

會輸出:

32

這樣,我們的Windows下安裝TensorFlow就完成了。

當(dāng)然,TensorFlow不僅僅支持Python語法,還有如下這些擴(kuò)展:

以上是"native" pip方式的安裝,另一種方式Anaconda大家也可以試下。不過pip是官方推薦的,Anaconda不是官方支持的。

最后,通過國內(nèi)阿里云鏡像看下TensorFlow的相關(guān)吧。

接下來你就可以運(yùn)用TensorFlow進(jìn)行相關(guān)數(shù)據(jù)訓(xùn)練和深度學(xué)習(xí)了。

后續(xù)將進(jìn)行TensorFlow的應(yīng)用講解...敬請期待..



以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號