在windows10上搭建caffe

caffe環境的搭建一直是讓我最頭疼的，最近在Windows10上成功搭建了caffe，在此對搭建過程進行記錄。

安裝主要是按照caffe github上的安裝說明進行的，caffe的github主頁中readme中有window caffe，是針對windows的安裝說明。

由於caffe框架有不少依賴環境，因此要首先按照說明安裝好依賴環境，cuda的安裝和cudnn的配置就再也不一一說明了，版本號要與說明的保持一致。還有cmake和python，能夠經過在命令窗口輸入python和cmake --version驗證cmake和python是否已經加入環境變量。如下命令均在命令窗口執行。

 

1. 下載caffe，經過git clone https://github.com/BVLC/caffe.git下載caffe到本地

2. 修改配置文件。

   • 經過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命令。

3. 配置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某個庫時出現了問題，而後從新安裝該庫就能夠了。