Caffe,全稱Convolution Architecture For Feature Extraction,是一個(gè)清晰且快速的深度學(xué)習(xí)框架。
下面介紹一下如何在Windows下配置Caffe框架
Caffe官網(wǎng)
Github: BVLC/caffe
官方的安裝說明很坑,自己總結(jié)一篇安裝過程造福后人。
預(yù)先準(zhǔn)備
編譯環(huán)境:Visual Studio 2013 Visual Studio 2015
一定不能用2015!
GPU工具:CUDA 8.0 下載完后把-exe
改為.exe
,直接安裝即可。
Python 2.7
cuDNN x64 把鏈接復(fù)制到迅雷等下載工具中即可,否則就注冊(cè)賬號(hào),按官網(wǎng)提示下載。
https://developer.nvidia.com/cudnn
之后我們到微軟在Github上提供的Caffe下載頁:Microsoft/caffe
下載好了ZIP文件后,解壓到任何位置,將文件夾改名為caffe
編譯配置
1.打開caffe文件夾,再打開windows文件夾,將其中CommonSettings.props.example
文件復(fù)制一份,改名為CommonSettings.props
2.解壓cudnn-8.0-windows10-x64-v5.1.1.zip
壓縮包,得到cuda
文件夾,將其復(fù)制到caffe
文件夾下
3.接下來我們編輯配置文件。右擊CommonSettings.props
,使用記事本或者其它文本編輯器打開。
將其中CudaVersion后面的7.5改為8.0
CuDnn是指定之前cuda
文件夾的路徑,我把caffe文件夾放在了D:\Workspace
下,因此這里填了D:\Workspace\caffe
4.找到路徑C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\extras\visual_studio_integration\MSBuildExtensions
文件夾,將其中的所有文件復(fù)制到C:\Program Files (x86)\MSBuild\Microsoft.Cpp\v4.0\V120\BuildCustomizations
中
5.這時(shí)候再打開Caffe.sln,就可以全部正常加載了。
開始編譯
1.右擊libcaffe,選擇Properties
2.屬性設(shè)置為如圖所示
3.之后再把libcaffe設(shè)置為啟動(dòng)項(xiàng)目
同時(shí)改上面編譯為Release
4.構(gòu)建libcaffe
這一過程會(huì)比較漫長,因?yàn)镹uGet會(huì)從網(wǎng)上把所有依賴文件全部下載到本地。大概有1個(gè)G左右。在Workspace文件夾下可以看到NuGetPackages文件夾,里面就是編譯用到的包。
注意VS的輸出窗口,看看Build是不是Succeeded了。
編譯好了后會(huì)出現(xiàn)一個(gè)Build的文件夾,進(jìn)caffe\Build\x64\Release
看看是否有相關(guān)文件
5.此時(shí),同理可以編譯caffe
也可以直接編譯Solution,把所有caffe的組件全部編譯好。
- 在Release文件夾里,就可以看到編譯好的caffe了
配置好了之后,就可以開始搞事情了~
參考:
Windows+VS2013爆詳細(xì)Caffe編譯安裝教程
Fing
2016.12.16