本教程是caffe在windows系統上的配置方法,而且是github上官方BVLC/caffe目前推薦的配置方法,並不是使用微軟的caffe-windows。
前期準備
- 首先從BVLC/caffe上下載caffe源碼,選擇好路徑後下載。下載過後,會得到一個caffe-windows.zip文件。
- 將該文件解壓到你指定的位置,進行下一步。
- 確定你是不是有NVIDIA的GPU,需不需要用GPU加速,如果不用GPU加速的話,可以跳過下面的cuda安裝和cudnn配置,直接配置caffe;如果需要用GPU,則需要在NVIDIA官網上分別下載安裝cuda和cdunn,下載地址在下面給出。
- 按照官方建議,需要使用cmake,所以需要先下載 cmake並安裝,給出的這個cmake鏈接是64位的,直接下載安裝,全英文界面,所有選項保持默認即可,一直點擊下一步,直到安裝完成。
環境配置
cuda和cudnn配置(不是使用NVIDIA GPU加速的話請掠過)
cuda和cudnn都是NVIDIA(英偉達)公司提供給開發者使用的運算平臺,簡而言之就是用GPU幹你想幹的事。
- 下載並安裝cuda,一定要選擇對應的版本,要不然會安裝失敗的。現在我演示的是64位win7,所以就選擇對應的版本下載。安裝過程比較簡單,雙擊安裝,選項建議都默認(有些人不喜歡把東西放在C盤,建議還是放在C盤,畢竟是開發工具),一直點擊確認或者下一步,直至安裝成功即可。
- 下載cudnn,這是用來加速深度網絡訓練的。下載cudnn需要登陸,你可以直接用郵箱註冊一個,然後登陸下載就好了。
- 下載完後,你會發現,cudnn不是一個軟件,而是一個壓縮包,我下載的就是:cudnn-8.0-windows10-x64-v6.0.zip(有人私信說caffe還是沒有編譯成功,我看了一下發現是cudnn版本問題,前面的cuda安裝的是8.0,所以cudnn應該用的是cudnn-8.0-windows10-x64-v6.0.zip,而不是其他版本)。這個包需要解壓縮,得到一個名爲cuda的文件夾,爲了不與前面安裝的cuda混餚,用紅色表示。
- 配置cudnn。cudnn解壓後得到的cuda文件夾下有bin、include、lib三個子文件夾,現在需要將這三個文件夾下的東西分別放到之前cuda的安裝目錄下的對應文件夾下。之前已經安裝過cuda,cuda的安裝目錄默認爲:C:\Program Files\NVIDIA GPU Computing Toolkit\,現在就只需要:
-將cudnn解壓後得到的cuda文件夾下的bin下的cudnn64_6.dll(cudnn後面的數字可能不一樣,這就是版本問題,如果是cuda 8.0 建議搭配我這個版本的cudnn,否則可能會出現問題)文件拷貝到cuda安裝目錄下的bin目錄下;
-將cudnn解壓後得到的cuda文件夾下的lib文件夾下的x64文件夾下的cudnn.lib拷貝到cuda安裝目錄下的lib文件夾下的x64目錄下
-將cudnn解壓後得到的cuda文件夾下的include文件夾下的cudnn.h拷貝到cuda安裝目錄下include文件夾下
可能上面講的可能有點囉嗦,但是希望你們能看懂啦。