caffe環境的搭建一直是讓我最頭疼的,最近在Windows10上成功搭建了caffe,在此對搭建過程進行記錄。
安裝主要是按照caffe github上的安裝說明進行的,caffe的github主頁中readme中有window caffe,是針對windows的安裝說明。
因為caffe框架有很多依賴環境,所以要首先按照說明安裝好依賴環境,cuda的安裝和cudnn的配置就不再一一說明了,版本號要與說明的保持一致。還有cmake和python,可以通過在命令窗口輸入python和cmake --version驗證cmake和python是否已經加入環境變量。以下命令均在命令窗口執行。
-
下載caffe,通過git clone https://github.com/BVLC/caffe.git下載caffe到本地
-
修改配置文件。
-
通過cd caffe進入caffe
-
通過git checkout windows進入windows分支
-
接下來就是修改scripts\build_win.cmd文件了,我需要安裝GPU版本的,需要配置python接口,下面是我進行的修改:
-
第8行,將WITH_NINJA置為0
-
第9行,將CPU_ONLY置為0
-
第15行,將PYTHON_VERSION置為3,因為我安裝的python是3.5版本的,caffe支持python2.7和python3.5,如果版本不對會出現未知錯誤。
-
第25行,將CONDA_ROOT置為C:\Anaconda3,因為我將Anaconda3的安裝目錄為C:\Anaconda3
-
第29行,同第25行,將CONDA_ROOT置為C:\Anaconda3
-
第74行,將WITH_NINJA置為0
-
第87行,將PYTHON_VERSION置為3
-
第172行,插入-DCUDNN_ROOT=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,因為cudnn配置到了cuda8.0中,這是它的路徑,當然也可以放上cudnn的原路徑
-
-
然后執行scripts\build_win.cmd命令,執行這一命令時,如果網絡有問題,可能會出錯。如果這一步執行正確,在caffe\build\tools\Release目錄中執行caffe命令,會得到下面的結果。
-
如果執行錯誤,查找錯誤原因,修正后先清除生成的build文件夾,然后重新執行scripts\build_win.cmd命令。
-
-
配置python接口,分別執行以下三條命令。這三條命令執行成功之后,將caffe\python\caffe文件夾拷貝到python的site-package中,在我的電腦中,我將其拷貝到C:\Anaconda3\Lib\site-packages中,這個根據個人python的安裝目錄進行修改。至此,python接口就配置好了,進入python,可以成功執行import caffe。
- conda config --add channels conda-forge
- conda config --add channels willyd
- conda install --yes cmake ninja numpy scripy protobuf==3.1.0 six scikit-image pyyaml pydotplus graphviz
我不需要配置matlab接口,至此,windows10系統中的caffe框架已經搭建完成。
我在配置python接口時,出現了問題,在此記錄一下以供參考,在執行完配置python接口的三條命令后,import caffe時出現了DLL not found的情況,根據錯誤信息,逐行測試,查到import某個庫時出現了問題,然后重新安裝該庫就可以了。