CAFFE 環(huán)境配置教程(win10+vs2015)
說明:本教程適用于在windoows平臺(tái)下搭建caffe環(huán)境有滑,搭建環(huán)境前請確認(rèn)自己的電腦系統(tǒng)為win10,有N卡GPU魂奥。
1.安裝VS2015
在微軟官網(wǎng)上下載vs2015的社區(qū)免費(fèi)版并安裝屋确,之前安裝過其他版本的VS不用卸載。
2.下載安裝CUDA8.0
根據(jù)自己的電腦配置選擇相應(yīng)的下載包,盡量選擇下載本地安裝包媳维,在線安裝包可能會(huì)出現(xiàn)小錯(cuò)誤底扳。
3.下載安裝CUDNN v5.0
注意這里的CUDNN版本铸抑,若下載其他版本的CUDNN可能與CUDA8.0不兼容
下載cuDNN v5(官網(wǎng)下載相應(yīng)的系統(tǒng)版本),將解壓出的三個(gè)文件夾剪切到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0覆蓋相應(yīng)文件夾衷模。
4.下載caffe的第三方支持包thirdparty
將百度云中的thirdparty壓縮包解壓到 caffe-windows-ms\windows\thirdparty并將caffe-windows-ms\windows\thirdparty\bins的完整路徑添加到path環(huán)境變量鹊汛。
5.將mnist.zip解壓到路徑caffe-windows-ms\data\mnist
批處理文件create_mnist-lmdb.bat 和 train_lenet.bat 復(fù)制到
caffe-windows-ms\examples\mnist
6.將caffe-windows-ms/windows/中的文件CommonSettings.props.example重命名為CommonSettings.props蒲赂,打開GPU Compute Capability查找對(duì)應(yīng)的顯卡計(jì)算容量, 3.0的需要將CommonSettings.props(用寫字板打開)中的
<CudaArchitecture>compute_52,sm_52;compute_60,sm_60</CudaArchitecture>
改為
<CudaArchitecture>compute_30,sm_30;compute_52,sm_52</CudaArchitecture>
其他顯卡改成
<CudaArchitecture>compute_35,sm_35;compute_52,sm_52</CudaArchitecture>
7.需要python支持的往下拉找python支持柒昏,建議盡量安裝這個(gè)凳宙。
(筆者當(dāng)時(shí)沒有安裝Python相關(guān)配置,導(dǎo)致LIBCAFFE中的一個(gè)Python相關(guān)的文件編譯失斨暗弧)
確認(rèn)CommonSettings.props中部分配置如下氏涩。
修改
<MXNetSupport>false</MXNetSupport>
修改props配置文件中的CudnnPath和PythonDir
8.用VS打開caffe-windows-ms\windows\caffe.sln,設(shè)置release模式有梆。
若提示需要加載相關(guān)組件如Python組件是尖,則按提示操作
9.對(duì)libcaffe項(xiàng)目右鍵生成,生成成功后再生成caffe.sln的解決方案
10.測試環(huán)境是否配置成功
caffe-windows-ms\examples\mnist\create_mnist-lmdb.bat
制作數(shù)據(jù)泥耀,train_lenet.bat訓(xùn)練數(shù)據(jù)饺汹,可以編輯更改
(結(jié)果日志應(yīng)該在C:\Users\用戶名\AppData\Local\Temp內(nèi)以“caffe.exe.”開頭)
訓(xùn)練截圖如下。
PYTHON支持安裝
- 安裝anaconda python 27版本痰催,官網(wǎng)免費(fèi)下載安裝兜辞。
- 進(jìn)入CMD運(yùn)行下面代碼
conda install protobuf
如安裝過其他Python可以使用
pip install protobuf
- 添加環(huán)境變量PythonDir指向anaconda安裝目錄
修改.\windows\CommonSettings.props 中的python 路徑<PythonDir> C:\Users\xxxx\Anaconda2\</PythonDir>
指向anaconda安裝目錄。 - 繼續(xù)安裝教程第8步
- 安裝完成后添加環(huán)境變量夸溶,“PythonPath” 指向 caffe_root\Build\x64\Release\pycaffe
- 進(jìn)入cmd依次輸入python ,import caffe測試是否通過