win8+theano+gpu加速

因?yàn)橐?xùn)練RNN(Recurrent Neural Network),所以想在windows上面安裝Theano,同時(shí)進(jìn)行GPU加速。促使我在deep learning的包里面選擇Theano的原因,是因?yàn)門heano對(duì)硬件的要求沒有那么高,同時(shí)也有很高的刻擴(kuò)展性。

其中windows8.1 64bit環(huán)境下搭建深度學(xué)習(xí)平臺(tái)主要參考win8 64bit平臺(tái)theano環(huán)境構(gòu)建(上)win8 64bit平臺(tái)theano環(huán)境構(gòu)建(下),Theano學(xué)習(xí)筆記:Theano的艱辛安裝體驗(yàn)這三個(gè)網(wǎng)頁(yè)上的文章。

安裝步驟:

一、首先安裝Visual Studio 2010或以上版本。

二、安裝Theano,現(xiàn)在實(shí)踐得出的比較簡(jiǎn)單的安裝方法是通過(guò)Anaconda安裝python來(lái)構(gòu)建Theano。

1) 首先下載Anoconda,選擇64bit版本。最好安裝在C:盤。

2) Anoconda安裝完以后在系統(tǒng)添加環(huán)境變量。我的電腦--右鍵單擊--屬性,然后在對(duì)話框左側(cè)選擇 高級(jí)系統(tǒng)設(shè)置,如下圖所示。

在跳出的對(duì)話框中繼續(xù)選擇 環(huán)境變量。

在跳出的對(duì)話框中,在系統(tǒng)變量(S)部分,選擇變量中的path選項(xiàng),然后將需要添加的環(huán)境變量添加進(jìn)去。這一部分我們需要添加的環(huán)境變量為:

C:\Anaconda2\MinGW\bin;?

C:\Anaconda2\MinGW\x86_64-w64-mingw32\lib;?

C:\Program Files (x86)\Microsoft Visual Studio 10.0\VC\bin;?

C:\Program Files (x86)\Microsoft Visual Studio 10.0\VC\bin\amd64

3) 然后Anaconda-2.0.1安裝完成后,在cmd命令下進(jìn)行Theano的安裝,Windows8.1中按快捷鍵"Win+R"出現(xiàn)對(duì)話框,輸入"cmd"即可進(jìn)入cmd對(duì)話框,然后執(zhí)行"pip install Theano"命令即可完成Theano的安裝。

看到提示"Successfully installed Theano"后,到Anaconda的安裝路徑下即可看到theano的文件夾,如下圖:

在cmd對(duì)話框里編寫一測(cè)試用例,即依次輸入如下命令,如果沒有報(bào)錯(cuò),說(shuō)明平臺(tái)搭建成功。

4) 新建環(huán)境變量:PYTHONPATH: C:\Anaconda2\Lib\site-packages\theano

三、安裝Cuda

1) 首先檢查自己的顯卡是否支持GPU并行運(yùn)算。檢查自己的nvidia顯卡是否在列表中。

2) 下載cuda toolkit并安裝。如果安裝的python是64bit的,那么cuda也應(yīng)該下載64bit的。我選擇的是cuda 64bit桌面版。文件下載好之后,直接雙擊就可以安裝。但是要注意安裝的過(guò)程中選擇自定義,而不是精簡(jiǎn)版。cuda安裝也可以參考cuda安裝參考。

3) cuda安裝好之后,也要添加對(duì)應(yīng)的環(huán)境變量。cuda安裝的過(guò)程中會(huì)自動(dòng)添加以下兩個(gè)環(huán)境變量。(我安裝的是cuda5.5的版本)

CUDA_PATH? ? ? ? ? ? ? C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v5.5

CUDA_PATH_V5_5 ? ? ?C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v5.5

我們需要手動(dòng)添加以下環(huán)境變量:

CUDA_BIN_PATH? ? %CUDA_PATH%\bin

CUDA_LIB_PATH? ? %CUDA_PATH%\lib\Win32

CUDA_SDK_PATH ? C:\ProgramData\NVIDIA Corporation\CUDA Samples\v5.5

CUDA_SDK_BIN? ? ? %CUDA_SDK_PATH%\bin\win32

CUDA_SDK_LIB ? ? ?%CUDA_SDK_PATH%\common\lib\win32

至此cuda的安裝完成。

3) 測(cè)試一下cuda的安裝是否成功。這個(gè)步驟我們用到兩個(gè)東西,這兩個(gè)東西,都是cuda為我們準(zhǔn)備好的。deviceQuery.exe 和 bandwidthTest.exe。

首先啟動(dòng)cmd DOS命令窗口默認(rèn)進(jìn)來(lái)的是c:\users\Admistrator\>路徑,輸入 cd .. 兩次,來(lái)到c:目錄下,輸入dir 找到安裝的cuda文件夾。我們根據(jù)下面的路徑進(jìn)入bandwidthTest.exe和deviceQuery.exe所在的文件夾。

然后我們運(yùn)行bandwidthTest.exe,得到下面的結(jié)果就表示通過(guò)了。

然后我們運(yùn)行deviceQuery.exe,得到下面的結(jié)果就表示通過(guò)了。

四、如果Theano和Cuda都安裝成功了,我們就配置一下Theano的設(shè)置,實(shí)現(xiàn)Theano通過(guò)GPU加速。我們需要寫一個(gè).theano的文件,然后放在c:\users\Admistrator\>路徑下。

五、我們?cè)趯?shí)際使用gpu進(jìn)行運(yùn)算的時(shí)候,可能會(huì)出現(xiàn)下面的情況。這里面顯示的問(wèn)題是CNMeM is disabled, cuDNN not available。

我們可以在.theano的配置文件中加上下面兩句,就可以實(shí)現(xiàn)CNMeM enabled,加快GPU的運(yùn)算速度。

[lib]

cnmem = 0.75

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容