本文在windows下使用visual studio2013配置關(guān)聯(lián)python(python-2.7.12.amd64.msi)的caffe項(xiàng)目,如果有耐心的人,當(dāng)然可以自己去下載caffe項(xiàng)目自己進(jìn)行編譯,由于我比較懶,所以這里我利用微軟自己移植的caffe進(jìn)行部署,就不嘗試編譯官方的caffe(聽(tīng)說(shuō)自己編譯需要解決各種坑),微軟移植的解決方案編譯確實(shí)很方便,解決一些引用就ok了。

 

(一)Caffe微軟官方GitHub

下載地址:https://github.com/Microsoft/caffe

通過(guò)GitHub管理項(xiàng)目非常方便,可以自己通過(guò)VS團(tuán)隊(duì)資源管理器進(jìn)行鏈接代碼,牛人修改里面的東西也可以提交上去。當(dāng)時(shí)我這里是直接下載caffe-master.zip的包的方式,然后直接解壓出來(lái),可以看到里面有28個(gè)文件,共包含16個(gè)項(xiàng)目:

photoshop培訓(xùn),電腦培訓(xùn),電腦維修培訓(xùn),移動(dòng)軟件開(kāi)發(fā)培訓(xùn),網(wǎng)站設(shè)計(jì)培訓(xùn),網(wǎng)站建設(shè)培訓(xùn)

(二)項(xiàng)目編譯前的步驟:

(1)設(shè)置配置文件

在 .\caffe-mastter\windows路徑下新建一個(gè)文本文檔,重命名為CommonSettings.props ,并復(fù)制.\windows\CommonSettings.props.example 文件中的內(nèi)容到 .\windows\CommonSettings.props。默認(rèn)情況下windows編譯需要CUDA和cuDNN,當(dāng)然你也可以在配置文檔(.\windows\CommonSettings.props)中去修改為不用,直接將為true的配置設(shè)置為false即可。但是python默認(rèn)情況下是不能使用的,我這里需要用python進(jìn)行caffe學(xué)習(xí),所以就設(shè)置為true了。其他三方的庫(kù),VS編譯時(shí)會(huì)通過(guò)NuGet進(jìn)行自動(dòng)獲取,非常方便!

為了便于在訓(xùn)練模型時(shí)減少輸出信息,建議去\caffe-master\caffe-master\src\caffe\layers\data_layer.cpp文件,將如下打印信息的代碼用"//"注釋掉:

DLOG(INFO) << "Prefetch batch: " << batch_timer.MilliSeconds() << " ms.";

DLOG(INFO) << "     Read time: " << read_time / 1000 << " ms.";

DLOG(INFO) << "Transform time: " << trans_time / 1000 << " ms.";

(2)下載相關(guān)軟件

CUDA

下載 CUDA Toolkit 7.5 (https://developer.nvidia.com/cuda-toolkit)。如果你電腦沒(méi)有NVIDIA的獨(dú)立顯卡,那么只能選擇用CPU進(jìn)行編譯,就不需要安裝CUDA,去配置文件 .\windows\CommonSettings.props 設(shè)置<CpuOnlyBuild>false</CpuOnlyBuild&

網(wǎng)友評(píng)論