天天看點

CAFFE FOR WINDOWS安裝與配置

CAFFE FOR WINDOWS安裝與配置

參照github上happynear的caffe-windows項目,将Caffe for Windows配置成功,并且測試了其轉換好的mnist資料庫,不論是速度還是結果上,效果都相當好。現總結一下配置方法。

實驗環境

作業系統 Windows 10 Professional
CPU Intel Core i5-4590
GPU Nvidia GeForce GTX 970
VS Microsoft Visual Studio 2013
CUDA CUDA 7.5

軟體安裝

首先需要安裝Visual Studio 2013。

然後再安裝CUDA 7.5。注意先後順序。

配置

下載下傳整個的caffe-windows項目,解壓縮到所需目錄下,例如本文中 D:\caffe。

CAFFE FOR WINDOWS安裝與配置

下載下傳作者制作的三方庫并解壓縮到項目的3rdparty目錄。解壓好後,将3rdparty/bin檔案夾加入到環境變量PATH中,這樣程式才能找到這些三方dll。

編譯

  1. 輕按兩下./src/caffe/proto/extractproto.bat批處理檔案來生成caffe.pb.h和caffe.pb.cc兩個c++檔案,和caffepb2.py這個python使用的檔案。
  2. 打開./buildVS2013/MainBuilder.sln,打開之後切換編譯模式至Release X64模式。
  3. 修改設定中的compute capability(caffelib –> 屬性 –> CUDA C/C++ –> Device –> Code Generation)
GPU Computer Capability
GTX660, 680, 760, 770 compute_30,sm_30
GTX780, Titan Z, Titan Black, K20, K40 compute_35,sm_35
GTX960, 970, 980, Titan X compute_52,sm_52
CAFFE FOR WINDOWS安裝與配置

你可以在https://en.wikipedia.org/wiki/CUDA上檢視你的GPU所對應的Computer Capability。

點選工具欄綠色箭頭進行編譯,需要一定時間。

CAFFE FOR WINDOWS安裝與配置

測試

下載下傳作者已經轉換好的MNIST的leveldb資料檔案,解壓到./examples/mnist檔案夾中,然後運作根目錄下的run_mnist.bat即可開始訓練,日志會儲存在./log檔案夾中。可看到,疊代10000次,準确率達到了0.9925。

CAFFE FOR WINDOWS安裝與配置

繼續閱讀