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阅读 218,204评论 6 506
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,091评论 3 395
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 164,548评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,657评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,689评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,554评论 1 305
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,302评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,216评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,661评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,851评论 3 336
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,977评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,697评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,306评论 3 330
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,898评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,019评论 1 270
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,138评论 3 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,927评论 2 355

推荐阅读更多精彩内容