作者:譚東
時間: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)用講解...敬請期待..
更多建議: