caffe第一步,配置VS2015環境。

      相信很多人都使用caffe,caffe確實是一個非常方便的框架,並且可以幫助大家很快的掌握深度學習框架,但是它的環境配置實在是痛苦萬分,現在給大家總結一下配置過程,希望不用走彎路。

      首先應該先做前期的準備工作,對此caffe官網的答案是:

可以根據以上需要下載準備軟件,CMake,VS2015,CUDA,cuDNN等。

1.下載Anaconda,以後直接conda安裝python包很方便 
我這裏用的是Anaconda2-4.3.1-Windows-x86_64版本。 
https://www.continuum.io/downloads

2.Cmake,一個跨平臺的安裝(編譯)工具,我用的是cmake-3.8.0-win64-x64。 
https://cmake.org/download/

3.CUDA,NVIDIA推出的運算平臺。本機使用CUDA8.0。

 https://developer.nvidia.com/cuda-downloads 

4.CUDNN,用於GPU加速的,本機使用V5.1 
https://developer.nvidia.com/cudnn 

有些人會問,其他版本的行嗎?我想說的是,不行,因爲caffe配置文件裏就聲明瞭,只支持4or5:

到這裏便做好了準備工作了。

接下來,我們需要去caffe官網下載caffe源碼,網址爲:https://github.com/BVLC/caffe/tree/windows

下載完畢後解壓,打開文件夾,找到scripts子文件夾

然後找到文本中的else可以看到許多選項,    1.MSVC_VERSION是編譯器版本,這裏因爲是要使用vs2015所以選擇的是14。     2.WITH_NINJA這裏默認爲1,我們將其改爲0,即不使用Ninja編譯器。     3.CPU_ONLY因爲我們要使用cuda,所以選擇0。    4.PYTHON_VERSION如果使用3.x的版本,這需要改爲3。其餘選項使用默認即可。

同時在往下移動,找到如圖所示地方

添加圖中,紅色部分的內容,其是你的cuDNN路徑。當配置好一切後。

請執行build_win.cmd(有些時候,由於網絡環境不佳,可能會遇到依賴包無法下載的問題,這時可以使用一些小技巧清除這個問題。具體方法是:手動下載包【https://github.com/willyd/caffe-builder/releases/download】,然後由於之前雖然下載的代碼執行失敗了可還是爲我們建立好了存放依賴庫的文件夾,我們把對應文件下載好後解壓放進去(默認是在C盤當前用戶目錄下的dependencies文件夾,download存放下載文件下來的文件,libraries_v140_x64_py27_1.1.0文件夾下則是存放該文件解壓後的library包

然後手動修改cmake文件夾下WindowsDownloadPrebuiltDependencies.cmake文件和scripts文件夾下download_prebuilt_dependencies.py文件代碼,讓程序繞過文件中下載和解壓的部分直接編譯,再執行build_win.cmd即可),

大約半小時後,將生成vs2015工程文件。

打開後,點擊生成ALL_BUILD生成結果即可。



發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章