雖然網上的教程很多,但是安裝的時候總會遇到不同的問題。是以記錄一下自己的安裝過程,希望能幫助到更多的人。
首先是必要的軟體和工具:
1、VS2015 我用的是community版
2、Python3.5 https://www.python.org/
3、Cmake(安裝3.4以上版本) https://cmake.org/download/
4、cuda8.0(目前8.0隻能是跟VS2015配合安裝) https://developer.nvidia.com/cuda-80-ga2-download-archive
5、cuDNN v5(注意版本,好像是v6以上會有錯誤) 需要注冊NVIDIA的賬号下載下傳https://developer.nvidia.com/cudnn
6、caffe官方的包(用git或者下載下傳zip檔案都可以) https://github.com/BVLC/caffe/tree/windows
7、Ninja https://ninja-build.org/ (下載下傳好後将exe檔案放入cmake的bin檔案夾)
以上都是用的64位
安裝完Python和Cmake以後要将bin所在的檔案夾添加到系統環境變量,具體操作如下:
這兩個路徑添加到我的電腦-屬性-進階系統設定-進階-環境變量-系統變量-輕按兩下path-加;添加路徑
接下來安裝好CUDA8.0 然後把cuDNN下載下傳好的壓縮包解壓,将三個檔案夾bin lib 和include放到CUDA的檔案夾内
這裡project是準備放caffe的檔案夾,可以自行調整路徑。如果不是下載下傳zip檔案的就用第一行代碼下載下傳caffe安裝包。用zip的直接解壓到指定檔案夾就行了。
在執行C:\Projects\caffe> scripts\build_win.cmd 之前,可以用notepad++修改一下參數。
用notepad打開 scripts\build_win.cmd 檔案, 第九行參數改為0,使用CUDA
if NOT DEFINED CPU_ONLY set CPU_ONLY=0
然後在cmd中輸入代碼
C:\Projects> git clone https://github.com/BVLC/caffe.git
C:\Projects> cd caffe
C:\Projects\caffe> git checkout windows
C:\Projects\caffe> scripts\build_win.cmd
開始安裝(漫長的過程 我用了20分鐘左右)(中間一個包的下載下傳出了點問題,重試了幾次)
編譯完成後,在caffe的build檔案夾裡我們可以看到編譯出來的檔案。用VS2015打開caffe.sln,開始編譯
編譯完成就安裝完成了