在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某個庫時出現了問題,然后重新安裝該庫就可以了。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM