Windows10下Caffe+CUDA+cuDNN+matcaffe+pycaffe安裝

GitHub Windows版Caffe分支 Windows Caffe

安裝環境

cuDNN安裝

  1. 安裝CUDA
  2. 下載cuDNN并解壓壓縮包
  3. 將解壓后的文件夾cuda下的文件分別復制到CUDA安裝目錄
cuDNN目錄 CUDA安裝目錄
cuda\bin C:\Program Files\NVIDIA GPU Computing Tookit\CUDA\v8.0\bin
cuda\include C:\Program Files\NVIDIA GPU Computing Tookit\CUDA\v8.0\include
cuda\lib\x64 C:\Program Files\NVIDIA GPU Computing Tookit\CUDA\v8.0\lib\x64

安裝Caffe

  1. D盤新建目錄CaffeBuild
  2. 打開cmd(命令提示符)切換到CaffeBuild目錄
> d:
> cd CaffeBuild
  1. 下載Caffe
D:\CaffeBuild> git clone https://github.com/BVLC/caffe.git
D:\CaffeBuild> cd caffe
D:\CaffeBuild> git checkout windows
  1. 修改配置
    D:\CaffeBuild\caffe\scripts下修改build_win.cmd文件,使用Sublime打開
    第8,9,14行
:: Default values
if DEFINED APPVEYOR (
    echo Setting Appveyor defaults
    if NOT DEFINED MSVC_VERSION set MSVC_VERSION=14
    if NOT DEFINED WITH_NINJA set WITH_NINJA=0
    if NOT DEFINED CPU_ONLY set CPU_ONLY=0
    if NOT DEFINED CUDA_ARCH_NAME set CUDA_ARCH_NAME=Auto
    if NOT DEFINED CMAKE_CONFIG set CMAKE_CONFIG=Release
    if NOT DEFINED USE_NCCL set USE_NCCL=0
    if NOT DEFINED CMAKE_BUILD_SHARED_LIBS set CMAKE_BUILD_SHARED_LIBS=0
    if NOT DEFINED PYTHON_VERSION set PYTHON_VERSION=3
    if NOT DEFINED BUILD_PYTHON set BUILD_PYTHON=1
    if NOT DEFINED BUILD_PYTHON_LAYER set BUILD_PYTHON_LAYER=1
    if NOT DEFINED BUILD_MATLAB set BUILD_MATLAB=1
    if NOT DEFINED PYTHON_EXE set PYTHON_EXE=python
    if NOT DEFINED RUN_TESTS set RUN_TESTS=1
    if NOT DEFINED RUN_LINT set RUN_LINT=1
    if NOT DEFINED RUN_INSTALL set RUN_INSTALL=1

第29行

:: Set python 3.5 with conda as the default python
    if !PYTHON_VERSION! EQU 3 (
        set CONDA_ROOT=D:\Anaconda3
    )

第74行

:: Change to 1 to use Ninja generator (builds much faster)
    if NOT DEFINED WITH_NINJA set WITH_NINJA=0

第87行

:: Change to 3 if using python 3.5 (only 2.7 and 3.5 are supported)
    if NOT DEFINED PYTHON_VERSION set PYTHON_VERSION=3

第91行

:: Change these options for your needs.
    if NOT DEFINED BUILD_PYTHON set BUILD_PYTHON=1
    if NOT DEFINED BUILD_PYTHON_LAYER set BUILD_PYTHON_LAYER=1
    if NOT DEFINED BUILD_MATLAB set BUILD_MATLAB=1

第167行添加

:: Configure using cmake and using the caffe-builder dependencies
:: Add -DCUDNN_ROOT=C:/Projects/caffe/cudnn-8.0-windows10-x64-v5.1/cuda ^
:: below to use cuDNN
cmake -G"!CMAKE_GENERATOR!" ^
      -DBLAS=Open ^
      -DCMAKE_BUILD_TYPE:STRING=%CMAKE_CONFIG% ^
      -DBUILD_SHARED_LIBS:BOOL=%CMAKE_BUILD_SHARED_LIBS% ^
      -DBUILD_python:BOOL=%BUILD_PYTHON% ^
      -DBUILD_python_layer:BOOL=%BUILD_PYTHON_LAYER% ^
      -DBUILD_matlab:BOOL=%BUILD_MATLAB% ^
      -DCUDNN_ROOT=C:\Program Files\NVIDIA GPU Computiong Toolkit\CUDA\v8.0 ^
      -DCPU_ONLY:BOOL=%CPU_ONLY% ^
      -DCOPY_PREREQUISITES:BOOL=1 ^
      -DINSTALL_PREREQUISITES:BOOL=1 ^
      -DUSE_NCCL:BOOL=!USE_NCCL! ^
      -DCUDA_ARCH_NAME:STRING=%CUDA_ARCH_NAME% ^
      "%~dp0\.."
  1. 執行腳本
D:\CaffeBuild\caffe> scripts\build_win.cmd

耐心等待,希望別報錯:)

下載依賴包可能因為網絡原因會失敗
網盤下載并放到下面這個目錄下,其中users后面的路徑改成你電腦的用戶名
C:\Users\shuai\.caffe\dependencies\download
這個依賴包只適合這個環境,其他環境需要搞定網絡后重新運行腳本


#報錯
'"C:\Program Files (x86)\Microsoft Visual Studio 14.0\Common7\Tools\..\..\VC\vcvarsall.bat"' 不是內部或外部命令,也不是 可運行的程序
或批處理文件。
-- The C compiler identification is unknown
-- The CXX compiler identification is unknown
CMake Error at CMakeLists.txt:18 (project):
  No CMAKE_C_COMPILER could be found.

 CMake Error at CMakeLists.txt:18 (project):
  No CMAKE_CXX_COMPILER could be found.

-- Configuring incomplete, errors occurred!
See also "E:/CaffeBuild/caffe/build/CMakeFiles/CMakeOutput.log".
See also "E:/CaffeBuild/caffe/build/CMakeFiles/CMakeError.log".
ERROR: Configure failed

解決方法:打開VS2015安裝程序,選擇修改,勾選編程語言下的Visual C++

勾選Visual C++

裝了兩次都出現下面這個錯誤

#報錯
CMake Error at cmake/Utils.cmake:69 (string):   
string sub-command STRIP requires two arguments.

解決方法:修改caffe\cmake下Utils.cmake,第69行加引號

# Function merging lists of compiler flags to single string.
# Usage:
#   caffe_merge_flag_lists(out_variable <list1> [<list2>] [<list3>] ...)
function(caffe_merge_flag_lists out_var)
  set(__result "")
  foreach(__list ${ARGN})
    foreach(__flag ${${__list}})
      string(STRIP ${__flag} __flag)
      set(__result "${__result} ${__flag}")
    endforeach()
  endforeach()
  string(STRIP "${__result}" __result)
  set(${out_var} ${__result} PARENT_SCOPE)
endfunction()

如果安裝成功則在caffe\build\tools\Release下有可執行文件

Python接口

打開Anaconda下的Anaconda Prompt

conda config --add channels conda-forge
conda config --add channels willyd
conda install --yes cmake ninja numpy scipy protobuf==3.1.0 six scikit-image pyyaml pydotplus graphviz

E:\CaffeBuild\caffe\python下caffe文件夾復制到E:\Anaconda3\Lib\site-packages

在cmd中輸入python,執行import caffe,若沒有報錯,則Python接口成功配置

MATLAB接口

E:\caffeBuild\caffe\matlab目錄下MATLAB中運行>> caffe.run_tests()
E:\CaffeBuild\caffe\matlab\+caffe\private\Release下的caffe_mexw64復制到E:\CaffeBuild\caffe\matlab\+caffe\private
修改matlab+caffe\Net.m第72行

function delete (self)
    if self.isvalid
        caffe_('delete_net', self.hNet_self);
    end
end

下載模型,cmd在caffe根目錄下執行

python scripts\download_model_binary.py models\bvlc_reference_caffenet

打開MATLAB,打開E:\CaffeBuild\caffe\matlab\demo\classification_demo.m
命令行窗口執行

im = imread('E:\CaffeBuild\caffe\examples\images\cat.jpg');

執行classification_demo.m
在MATLAB命令窗口執行help caffe,如果不報錯,則MATLAB接口配置成功

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,505評論 6 533
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,556評論 3 418
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,463評論 0 376
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,009評論 1 312
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,778評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,218評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,281評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,436評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,969評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,795評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,993評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,537評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,229評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,659評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,917評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,687評論 3 392
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,990評論 2 374

推薦閱讀更多精彩內容