windows 10 下 Caffe + Matlab 部署

在 windows 10 下部署 Caffe 花了不少时间才调通,记录下 key 节点以便后续查询:html

 

1、安装软件:git

一、安装 Microsoft Virtual Studio 2013 / Matlab 2015a / CUDA 7.5:github

须要注意的是, VS2013 须要先安装,以便于 Matlab 识别 VS路径,CUDA 绑定内容;总之须要 VS2013 第一个安装;redis

其中,VS2013 SP5 是必须的版本,据 happynear [CSDNGitHub]描述,Caffe 在 VS2012 下编译速度很慢(自测 VS2013 约半小时,i7 6700HQ / 64G / 970M / 3G + SSD);windows

CUDA 测试用的 7.5 版本,其余版本的 Caffe 不知道是否须要修改相关源码;网络

Matlab 测试使用的是 2015a,在前一次重置系统前是 2015b,彷佛也没有异常,默认是 2014b,有欢迎留言反馈其余版本;app

二、安装 cuDNN:性能

cuDNN 改善了 Caffe native 卷基层效率不高的弱点(参见 赵永科/卜居 先生的大做《深度学习-21天实战caffe》一书 P356 所提到的卷积网络性能对比),笔者下载的分支为 Microsoft 为 windows 所开发的分支,支持 cuDNN V3/V4,对于最新版本为 V5,Caffe 还没有官方(Microsoft 分支)支持,如要支持请参考卜居先生的指导;若是没有 nVIDIA 帐号,须要注册一个帐号才能下载 cuDNN(网上搜能够获得网盘等资源,文件名:cudnn-7.0-win-x64-v4.0-prod.zip cudnn-7.0-win-x64-v3.0-prod.zip cudnn-7.5-windows10-x64-v5.0-ga.zip );学习

此外,建议 cuDNN 单独安装在一个目录中,独立于 CUDA,方便更换版本(删除就行);测试

三、安装 Direct X SDK [非必选]:

Direct X SDK 是非必须的,CUDA 的 Demo 编译时提示几个例程须要 d3dx9.h 文件,下载这个版本就好;

若是安装出错(非全新系统容易出现),提示“Error Code: S1023:Setup failed. Errors were encountered during installation of redistributable packages. Please close all open programs and try running setup again. If problems persist, contact DirectX Developer Support.”,错误以下图,能够卸载 Microsoft Visual C++ 2010 x86/x64 redistributable 后继续安装(参考文章)。 

四、安装 Miniconda 部署 [非必选]:

为了编译 Python 版本的 Caffe,须要安装 Miniconda,Microsoft/Caffe 的 READ.ME 里面有相关的说明;

 

2、编译 Caffe windows 分支:

开始以前,建议先编译一下 CUDA 自带的 Samples,以便测试 CUDA 环境,若是编译运行有错误,能够在这阶段方便地分离问题。

Caffe 的 windows 分支有不少版本,测试使用的是 windows 版,这一版须要修改的内容较少(Happynear 版尝试了下,太复杂不适合小白),仍是 Microsoft 的版本 Microsoft/caffe 比较容易,记录以下:

一、下载或 Git 源码:

地址:传送门

二、复制 .\windows\CommonSettings.props.example 文件,重命名为 .\windows\CommonSettings.props;

三、修改 cuDNN 使能选项,决定是否使用 cuDNN;

 三、安装 cuDNN 时,若是没有解压 cuDNN 在 CUDA 目录里面,须要修改 cuDNN 路径;

四、修改 Matlab 路径为实际路径:

五、能够打开 Caffe.sln 并准备编译了:

windows 版须要依赖 NuGet 提供的第三方库,因此电脑可以上网,且网速良好是十分必要的(该死的长城宽带 / 该死的长城宽带 / 该死的长城宽带);

编译可能会遇到一堆错误,没关系,一个个检查过去就行了;

 其中测试时遇到的有:

显示警告:“math_functions.h(4492): warning C4819: The file contains a character that cannot be represented in the current code page (936). Save the file in Unicode format to prevent data loss”,相似这样数百数千个警告,不用管就能够;

提示 alt_sstream_impl.hpp 文件有问题:“Error 8711 error C2220: warning treated as error - no 'object' file generated (..\..\src\caffe\util\math_functions.cpp) .\Caffe\NugetPackages \ boost.1.59.0.0\ lib \native \include \boost\format\alt_sstream_impl.hpp”,并提示了出错的行,这是这个文件的编码或是结尾等内容与当前系统或编译器设置不匹配,简单按下 Ctrl + S 重写 alt_sstream_impl.hpp 保存便可;

其余还可能遇到的错误,请FQ或百度一下(留言也能够,小白我不必定能解决);

须要注意的是,若是须要运行 Matlab,须要在编译前运行 MatlabPreBuild.cmd,编译后运行 MatlabPostBuild.cmd(没有深究编译先后是否自动调用了);

 

解决了必然存在的编译错误,就愉快地生成了 windows 版的所有文件;

 

3、跑 Matlab 版的测试:

一、下载基本数据:

能够在 GitHub 上下载须要的 bvlc_reference_caffenet.caffemodel 文件,或者寻找其余资源如网盘;下载完成后解压至 .\models\bvlc_reference_caffenet 目录;

 

二、准备环境变量:

添加编译出来的 .\Build\x64\Release 文件夹到环境变量的 path 中,若是运行不成功记得重启系统再试(很诡异,测试时重启才看到环境变量生效);

三、启动 Matlab,指向 .\matlab\demo 目录;

添加路径 .\Build\x64\Release\matcaffe 到 Matlab Path 中,而后运行以下命令: 

1

2

3

im = imread('../../examples/images/cat.jpg');

scores = classification_demo(im, 1);

[score, class] = max(scores);

获得输出:

1

2

3

Elapsed time is 0.073357 seconds.

Elapsed time is 0.153671 seconds.

Cleared 0 solvers and 1 stand-alone nets

能够看出,获得了最大可能的标签为 282,0.2985.

运行以下代码能够看到更多的分类结果细节:

1

2

3

figure;plot(scores);

axis([0, 999, -0.1, 0.5]);

grid on

如图:

 

通过艰难尝试,Caffe 的 windows 版和 Faster-RCNN 前后攻克,标记一下解决过程,以备后续查找。

 

此外,更常见的错误是 “Invalid MEX-file '……\caffe.mexw64': 找不到指定的模块”,这个错误须要分两步解决,第一步是添加环境变量,目录为 .\Build\x64\Release;第二步是将 matcaffe 加入 matlab 的目录中(可能非必须,不加也成功过);

修改系统环境变量时,测试失败,但重启就能够,因此加了变量还不行建议重启系统试试;若是一直不行,那么将 .\Build\x64\Release 下生成的所有文件都复制到 .\matlab\+caffe\private 下,重启 Matlab 重试,应该能够解决。

相关文章
相关标签/搜索