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

2018-09-01 21:25 更新

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

作者:譚東

時(shí)間:2017年5月28日

環(huán)境:Windows 7

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

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

給你們幾個(gè)關(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 不是一個(gè)嚴(yán)格的“神經(jīng)網(wǎng)絡(luò)”庫(kù)。只要你可以將你的計(jì)算表示為一個(gè)數(shù)據(jù)流圖,你就可以使用Tensorflow。

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

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

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

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

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

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

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

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

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

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

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

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

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

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

國(guó)內(nèi)PIP源:

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

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

中國(guó)科技大學(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)去下載對(duì)應(yīng)的Python Windows安裝包就行了,要注意32位和64位版本的選擇。

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

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

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

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

3.Python安裝好后,我們就要進(jìn)行下載安裝TesorFlow需要的庫(kù)了。

CPU版本命令輸入:

pip3 install --upgrade tensorflow

GPU版本命令輸入:

pip3 install --upgrade tensorflow-gpu

等待自動(dòng)下載。

整個(gè)過(guò)程TensorFlow需要安裝:numpy、six、wheel、appdirs、pyparsing、packaging、setuptools、protobuf、werkzeug、tensorflow。

4.安裝完TensorFlow后,我們寫一個(gè)例子進(jìn)行測(cè)試下是否可以運(yùn)行。在程序里找到:Python3.5或者CMD里輸入Python即可進(jìn)入輸入編碼模式。

給大家兩個(gè)例子:

官方例子:

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

會(huì)輸出:

Hello, TensorFlow!

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

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

會(huì)輸出:

32

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

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

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

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

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

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



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

掃描二維碼

下載編程獅App

公眾號(hào)
微信公眾號(hào)

編程獅公眾號(hào)