Caffe2-windows下caffe的gpu加速
摘要:本篇日志承載上篇內(nèi)容,在上篇基礎(chǔ)上進(jìn)行優(yōu)化和添加。主要說(shuō)明在正確cpu的caffe框架下如何添加gpu加速的過(guò)程。
正文:
???????? 上篇說(shuō)明了如何安裝cpu版本的caffe框架,其中我提到過(guò),只有cpu的caffe就好比沒(méi)有輪胎的車(chē),好賴(lài)是車(chē)子但是還沒(méi)發(fā)跑起來(lái)。雖然這樣,我們?nèi)匀灰呀?jīng)完成了框架搭載80%的內(nèi)容。這里再聲明一下對(duì)計(jì)算機(jī)硬件的要求:gtx750以上、顯存2G以上的N卡,i3以上的cpu,滿(mǎn)足這些條件可以繼續(xù)我們的工程了。
???????? 首先你需要在英偉達(dá)的官網(wǎng)下載cuda這里給出cuda的官網(wǎng):?https://developer.nvidia.com/cuda-downloads,因?yàn)榇税惭b包近乎1G就不附帶我的百度云地址。此處有必要強(qiáng)調(diào),cuda的版本是7.0或者8.0,windows版本,選擇頁(yè)面如圖:
???????? 下載好后選擇默認(rèn)安裝,remember默認(rèn)安裝,不要自作主張更改安裝路線。這個(gè)安裝會(huì)持續(xù)20min左右,在安裝此應(yīng)用期間不得運(yùn)行其他安裝軟件(這是此軟件的要求),當(dāng)cuda安裝完畢后還需要下載cuda的驅(qū)動(dòng),同樣給出驅(qū)動(dòng)的官網(wǎng),根據(jù)自己的計(jì)算機(jī)型號(hào)選擇驅(qū)動(dòng)類(lèi)型:http://www.nvidia.com/Download/index.aspx?lang=en-us。此處強(qiáng)調(diào),cuda驅(qū)動(dòng)和cuda不是一回事!不是一回事,驅(qū)動(dòng)大小約為350Mb左右,所以根據(jù)大小相信觀眾不會(huì)下載錯(cuò)誤,這是例子:
???????? 下載好驅(qū)動(dòng)后仍然默認(rèn)安裝,不可以更改安裝路線。
???????? 這里對(duì)上述兩個(gè)安裝過(guò)程中可能出現(xiàn)的問(wèn)題進(jìn)行解答:
1.?????????????可能出現(xiàn)的bug是“shadow 安裝失敗”這個(gè)bug,無(wú)須擔(dān)心,我們的計(jì)算不用到這個(gè)模塊。
2.?????????????可能出現(xiàn)一個(gè)叉號(hào)并且顯示“與以前版本沖突并且沒(méi)有更新”,這個(gè)問(wèn)題更好解決,就是你曾經(jīng)安裝的版本部分更新和此次安裝一樣,不需要重復(fù)更新。
安裝好兩個(gè)文件以后就完事具備了嗎?no~no~no~。接下來(lái)是很容易出現(xiàn)錯(cuò)誤的一步:添加環(huán)境變量,這是個(gè)經(jīng)常要用到的技術(shù),因此我在caffe日志1中做了詳細(xì)的圖解。當(dāng)cuda正確安裝后,在環(huán)境變量列表中會(huì)出現(xiàn)如下兩個(gè)變量:CUDA_PATH和CUDA_PATH_V8_0,剩余的需要我們自己去添加:
CUDA_SDK_PATH= C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0
CUDA_LIB_PATH= %CUDA_PATH%\lib\x64
CUDA_BIN_PATH= %CUDA_PATH%\bin
CUDA_SDK_BIN_PATH= %CUDA_SDK_PATH%\bin\win64
CUDA_SDK_LIB_PATH= %CUDA_SDK_PATH%\common\lib\x64
然后,在系統(tǒng)變量PATH的末尾添加:
;%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%;
此處需要強(qiáng)調(diào)的是:
1.??????系統(tǒng)變量path中注意每個(gè)變量之間以‘;’為分隔符;
2.??????變量大小寫(xiě)不影響正確性。
3.??????自己添加變量時(shí)的方式是名稱(chēng)一欄寫(xiě)變量名,如CUDA_BIN_PATH,變量一欄寫(xiě)地址,如%CUDA_SDK_PATH%/bin/64/win64,至于斜杠是/的還是\這是沒(méi)有關(guān)系的,不過(guò)和大眾一樣總沒(méi)錯(cuò)對(duì)吧。
如果正確完成了這一步,那么你的gpu配置就已經(jīng)完成了,那么你可能會(huì)問(wèn)我可以直接調(diào)用gpu加速計(jì)算了嗎?不可以,因?yàn)槟阒皇墙o系統(tǒng)配置了一個(gè)pgu,并沒(méi)有給caffe說(shuō)啊。下面說(shuō)明如何在caffe中添加gpu模式。
????? 再次打開(kāi)日志1中的配置文件,在cpu-only那里寫(xiě)false,這說(shuō)明你已經(jīng)調(diào)用了gpu了。
????? 是不是不很復(fù)雜呢?的確,如果想一步安裝caffe到成功,90%會(huì)失敗,而像這樣步步漸進(jìn)穩(wěn)中求進(jìn)是不是也很不錯(cuò)呢?現(xiàn)在我們的caffe已經(jīng)裝上一個(gè)輪胎啦,再裝三個(gè)輪子這輛車(chē)子就可以跑起來(lái)了!后續(xù)要說(shuō)明如何在里面安裝Python接口,以使得caffe可以訓(xùn)練數(shù)據(jù),如何接口MATLAB,以使得caffe可以可視化,以及如何用caffe訓(xùn)練,預(yù)測(cè),判斷,如何設(shè)計(jì)自己的神經(jīng)網(wǎng)絡(luò)等等。。。請(qǐng)聽(tīng)下回分解!
?
?
總結(jié)
以上是生活随笔為你收集整理的Caffe2-windows下caffe的gpu加速的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: [日常笔记] 2: 本地视频播放器 MP
- 下一篇: 简易实训教学授课系统解决方案