文將解釋如何在Windows下安裝TensorFlow。
確定安裝哪類(lèi)TensorFlow
需要先確定哪種類(lèi)型的TensorFlow:
運(yùn)行TensorFlow所需要的GPU配置
如果在系統(tǒng)中安裝使用GPU運(yùn)行的TensorFlow,需要確保下面介紹的NVIDIA軟件已經(jīng)安裝到系統(tǒng)中。
如果系統(tǒng)中已經(jīng)安裝了以前的相關(guān)包,請(qǐng)更新到所指定的版本。
如何安裝TensorFlow
在安裝TensorFlow之前必須選定一個(gè)安裝機(jī)制。目前提供2種機(jī)制:
Native的安裝(以下簡(jiǎn)稱(chēng)本地安裝)方式會(huì)將TensorFlow直接安裝在當(dāng)前的系統(tǒng)中,不會(huì)在系統(tǒng)和TensorFlow之間搭建任何的虛擬環(huán)境,所以本地安裝不會(huì)額外安裝一個(gè)獨(dú)立的容器。需要注意的是本地安裝可能會(huì)干擾系統(tǒng)中其他基于python安裝的程序。如果事先已經(jīng)安裝配置了滿(mǎn)足需要的python環(huán)境,本地安裝通常只需要一個(gè)命令就可以完成。使用本地安裝,用戶(hù)可以在系統(tǒng)中任何位置運(yùn)行TensorFlow。
在Anaconda模式下,需要使用conda創(chuàng)建一個(gè)虛擬環(huán)境。官方優(yōu)先推薦使用 pip install 命令來(lái)安裝TensorFlow,其次再考慮anaconda的 conda install 命令。conda包是第三方社區(qū)提供的(非TensorFlow官方),TensorFlow團(tuán)隊(duì)從始至終都不會(huì)去測(cè)試在conda中運(yùn)行的情況,在使用時(shí)需考慮這個(gè)風(fēng)險(xiǎn)。
本地安裝
首先,需要安裝以下版本的python:
TensorFlow在windows操作系統(tǒng)中僅僅支持3.5.x版本的python。Python 3.5.x附帶pip3軟件包管理器,這是用于安裝TensorFlow的程序。
安裝TensorFlow需要啟動(dòng)一個(gè)終端(terminal),然后在該終端中輸入對(duì)應(yīng)的pip3 install命令。安裝僅支持CPU版本的TensorFlow,輸入以下命令:
C:\> pip3 install --upgrade tensorflow
安裝GPU版本的TensorFlow,使用以下命令:
C:\> pip3 install --upgrade tensorflow-gpu
Anaconda模式安裝
再次強(qiáng)調(diào),Anaconda安裝是有第三方社區(qū)提供的,非官方。
在Anaconda環(huán)境中安裝TensorFlow分為以下幾個(gè)步驟:
C:> conda create -n tensorflow
C:> activate tensorflow (tensorflow)C:> # Your prompt should change
(tensorflow)C:> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.0.1-cp35-cp35m-win_amd64.whl
(tensorflow)C:> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.0.1-cp35-cp35m-win_amd64.whl
驗(yàn)證安裝
C:> python
>>> import tensorflow as tf hello=tf.constant('Hello, TensorFlow!') sess=tf.Session() print(sess.run(hello))
Hello, TensorFlow!
常見(jiàn)的安裝問(wèn)題
TensorFlow通過(guò)Stack Overflow網(wǎng)站來(lái)記錄錯(cuò)誤信息以及處理方法。下面的列表包含一些跳轉(zhuǎn)的到 Stack Overflow的連接。如果在安裝過(guò)程中遇到的問(wèn)題沒(méi)有在下面中,請(qǐng)到Stack Overflow去搜索相關(guān)的關(guān)鍵字。若還是搜索不到,請(qǐng)直接提出新問(wèn)題并標(biāo)記 tensorflow 的標(biāo)簽。
Stack Overflow LinkError Message41007279 [...\stream_executor\dso_loader.cc] Couldn't open CUDA library nvcuda.dll41007279 [...\stream_executor\cuda\cuda_dnn.cc] Unable to load cuDNN DSO42006320 ImportError: Traceback (most recent call last): File "...\tensorflow\core\framework\graph_pb2.py", line 6, in from google.protobuf import descriptor as _descriptor ImportError: cannot import name 'descriptor'42011070
No module named "pywrap_tensorflow"參考文獻(xiàn):K碼農(nóng)-http://kmanong.top/kmn/qxw/form/home?top_cate=28
近期由個(gè)人工作的需要,著手開(kāi)始研究深度學(xué)習(xí)相關(guān)的內(nèi)容。也是小白一枚,希望通過(guò)文字來(lái)記錄自己學(xué)習(xí)過(guò)程中踩的坑,給看到的人一點(diǎn)幫助。
Tensorflow
目前深度學(xué)習(xí)框架中,最火熱也是相關(guān)社區(qū)支持比較好的就是Google發(fā)布的Tensorflow。所以我開(kāi)始了對(duì)Tensorflow的學(xué)習(xí)。最早Tensorflow不支持windows,于是我使用的是Ubuntu 16.04進(jìn)行安裝,不過(guò)當(dāng)前版本已經(jīng)支持的windows系統(tǒng),所以記錄一下我安裝過(guò)程中的步驟和踩過(guò)的坑。
系統(tǒng):Win7 旗艦版64位(windows10版本親測(cè)可用)
可以直接安裝python,但小白一枚,安裝各種包比較麻煩,因此我直接裝了Anaconda集成環(huán)境。
下載好后打開(kāi)安裝包,可以更改默認(rèn)安裝路徑,勾選下圖的部分,其余一路無(wú)腦next,下圖兩個(gè)選項(xiàng)都勾選。
安裝時(shí)勾選這兩部分
安裝完成后win+R cmd可以查看是否安裝正常
檢查是否安裝正確
作為初學(xué)者,我們安裝cpu版本的tensorflow足夠使用。安裝TensorFlow-cpu很簡(jiǎn)單,以管理員權(quán)限打開(kāi)cmd,輸入:
pip install tensorflow
稍等一會(huì)兒即可安裝成功
查看是否安裝成功:
檢查是否安裝正確
以上是正常的安裝方法,但下載是及其慢的,可以自己感受下后來(lái)各種百度,發(fā)現(xiàn)還是得先提前下載好,完了指定路徑本地安裝是最快的,路徑是這樣的:
https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-0.12.0rc0-cp35-cp35m-win_amd64.whl
類(lèi)似這種,(網(wǎng)上找的,要改成你自己的whl包),
我的是tensorflow-1.4.0-cp36-cp36m-win_amd64.whl,所以應(yīng)該怎么寫(xiě):
https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.4.0-cp36-cp36m-win_amd64.whl
將連接直接復(fù)制到瀏覽器中。
下載到本地,例如我的路徑是f:\tensorflow\,打開(kāi)cmd,使用命令:
安裝過(guò)程
安裝速度很快,完了之后就可以按之前的測(cè)試方式驗(yàn)證是否安裝完成。
至此,windows安裝Tensorflow就結(jié)束了,大家可以開(kāi)始深度學(xué)習(xí)之旅了。