摘要：本篇日志承載上篇內(nèi)容，在上篇基礎(chǔ)上進行優(yōu)化和添加。主要說明在正確cpu的caffe框架下如何添加gpu加速的過程。

正文：

???????? 上篇說明了如何安裝cpu版本的caffe框架，其中我提到過，只有cpu的caffe就好比沒有輪胎的車，好賴是車子但是還沒發(fā)跑起來。雖然這樣，我們?nèi)匀灰呀?jīng)完成了框架搭載80%的內(nèi)容。這里再聲明一下對計算機硬件的要求：gtx750以上、顯存2G以上的N卡，i3以上的cpu，滿足這些條件可以繼續(xù)我們的工程了。

???????? 首先你需要在英偉達的官網(wǎng)下載cuda這里給出cuda的官網(wǎng):?https://developer.nvidia.com/cuda-downloads，因為此安裝包近乎1G就不附帶我的百度云地址。此處有必要強調(diào)，cuda的版本是7.0或者8.0，windows版本，選擇頁面如圖：

???????? 下載好后選擇默認安裝，remember默認安裝，不要自作主張更改安裝路線。這個安裝會持續(xù)20min左右，在安裝此應(yīng)用期間不得運行其他安裝軟件（這是此軟件的要求），當(dāng)cuda安裝完畢后還需要下載cuda的驅(qū)動，同樣給出驅(qū)動的官網(wǎng)，根據(jù)自己的計算機型號選擇驅(qū)動類型：http://www.nvidia.com/Download/index.aspx?lang=en-us。此處強調(diào)，cuda驅(qū)動和cuda不是一回事！不是一回事，驅(qū)動大小約為350Mb左右，所以根據(jù)大小相信觀眾不會下載錯誤，這是例子：

???????? 下載好驅(qū)動后仍然默認安裝，不可以更改安裝路線。

???????? 這里對上述兩個安裝過程中可能出現(xiàn)的問題進行解答：

1．?????????????可能出現(xiàn)的bug是“shadow 安裝失敗”這個bug，無須擔(dān)心，我們的計算不用到這個模塊。

2．?????????????可能出現(xiàn)一個叉號并且顯示“與以前版本沖突并且沒有更新”，這個問題更好解決，就是你曾經(jīng)安裝的版本部分更新和此次安裝一樣，不需要重復(fù)更新。

安裝好兩個文件以后就完事具備了嗎？no~no~no~。接下來是很容易出現(xiàn)錯誤的一步:添加環(huán)境變量，這是個經(jīng)常要用到的技術(shù)，因此我在caffe日志1中做了詳細的圖解。當(dāng)cuda正確安裝后，在環(huán)境變量列表中會出現(xiàn)如下兩個變量：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%;

此處需要強調(diào)的是：

1.??????系統(tǒng)變量path中注意每個變量之間以‘;’為分隔符；

2.??????變量大小寫不影響正確性。

3.??????自己添加變量時的方式是名稱一欄寫變量名，如CUDA_BIN_PATH，變量一欄寫地址，如%CUDA_SDK_PATH%/bin/64/win64，至于斜杠是/的還是\這是沒有關(guān)系的，不過和大眾一樣總沒錯對吧。

如果正確完成了這一步，那么你的gpu配置就已經(jīng)完成了，那么你可能會問我可以直接調(diào)用gpu加速計算了嗎？不可以，因為你只是給系統(tǒng)配置了一個pgu，并沒有給caffe說啊。下面說明如何在caffe中添加gpu模式。

????? 再次打開日志1中的配置文件，在cpu-only那里寫false，這說明你已經(jīng)調(diào)用了gpu了。

????? 是不是不很復(fù)雜呢？的確，如果想一步安裝caffe到成功，90%會失敗，而像這樣步步漸進穩(wěn)中求進是不是也很不錯呢？現(xiàn)在我們的caffe已經(jīng)裝上一個輪胎啦，再裝三個輪子這輛車子就可以跑起來了！后續(xù)要說明如何在里面安裝Python接口，以使得caffe可以訓(xùn)練數(shù)據(jù)，如何接口MATLAB，以使得caffe可以可視化，以及如何用caffe訓(xùn)練，預(yù)測，判斷，如何設(shè)計自己的神經(jīng)網(wǎng)絡(luò)等等。。。請聽下回分解！

?

?

總結(jié)

以上是生活随笔為你收集整理的Caffe2-windows下caffe的gpu加速的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。