Win10+VS2015安装Caffe(含Matlab和Python接口配置)

Win10下使用VS2015安装Caffe

本教程主要是在Win10下使用VS2015编译Caffe并配置Matlab2016b和Python3.5接口(注意:Pyhton版本只能是3.5或者2.7),最近看了很多Windows下编译Caffe的教程,大部分用的都是旧版本的包(好像官方GitHub上已经下载不到了),这里使用当前最新的Windows分支的Caffe来编译,遇到了许多坑,因此在这里做一些总结,希望能帮助到大家。

  • 1、安装Cmake3.12和VS2015
  • 2、安装CUDA8.0和cudnn6.0
  • 3、安装Matlab2016b和Python3.5
  • 4、修改编译脚本
  • 5、编译Caffe
  • 6、配置路径
  • 7、测试

1、安装Cmake3.12和VS2015

安装Cmake3.12和VS2015,Cmake版本3.4以上,VS2013或者VS2015都行(Cmake3.12),安装完成后找到Cmake的安装路径,将下面的bin文件夹路径添加到系统环境变量中去。

2、安装CUDA8.0和cudnn6.0

CUDA8.0和Cudnn6.0安装方法请自行百度。(如果配置CPU版本,请忽略此项)

3、安装Matlab2016b和Python3.5

如果需要配置MatLab接口请自行下载Matlab2016b(其他版本应该也行)安装,这里主要说一下怎么搭建Python3.5,不要直接安装python3.5,建议安装Anaconda3,然后新建一个python版本为3.5的环境(注意新版的anaconda默认的都是python3.6,所以需要自己新建一个3.5的环境,名称可以为caffe),将这里创建的python3.5的环境路径添加到系统环境变量中去,这一步很重要,添加完后在cmd中输入python测试一下,python版本是否为3.5(注意python可以用2.7版本,流程类似)然后打开Anaconda Prompt,进入刚刚创建的caffe环境(activate caffe 即可)运行以下代码,安装依赖包

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

4、修改编译脚本

下载Caffe的Windows编译包,微软给的官方代码包在这里(https://github.com/BVLC/caffe/tree/windows),如果本博客不算过时,建议使用这里的包(Caffe.rar)解压之后将Caffe文件夹放到D盘根目录,找到下面的scripts下面的build_win.cmd文件,使用文本编辑器打开,按以下方式修改(注意看图片中修改位置的行号)
这里写图片描述
这里写图片描述
这里写图片描述

5、编译Caffe

接下来就可以开始编译Caffe了,进入Caffe下的scripts文件夹下,按住Shift键,鼠标右键,点击“在此次打开Powershell窗口”,就会弹出命令框,在命令行中输入build_win.cmd然后回车,就会开始编译了,注意编译过程中可能会下载一下依赖文件,请保证网络情况良好,之后会有很多警告信息可以忽略,编译大约需要15-30min,请耐心等待。
这里写图片描述

6、配置路径

编译成功后会显示如下信息:
这里写图片描述
首先,将Caffe下的“D:\Caffe\scripts\build\tools\Release”(具体对应于自己电脑的路径)添加到系统环境变量中去,接下来配置Python接口路径,先将D盘Caffe下Python下的caffe文件夹复制到上面配置的Anaconda中的caffe环境下,例如我电脑上的路径是“D:\Anaconda3\envs\caffe\Lib\site-packages”,(注意这里这样主要是方便在Anaconda中caffe环境下使用Caffe)然后我们来配置Matlab的接口,先将“D:\Caffe\matlab+caffe\private\Release”下面的caffe_.mexw64文件拷贝到它上一级目录“D:\Caffe\matlab+caffe\private”中去,然后打开Matlab
这里写图片描述
添加文件夹,找到“D:\Caffe\matlab”添加保存即可。

7、测试

首先测试Matlib接口,打开matlab,在命令行窗口输入caffe.run_tests()如果没有报错则表示matlab接口配置成功(注意可能会出现“警告: 执行 ‘caffe.Solver’ 类析构函数时,捕获到以下错误:”等信息,这里不影响使用)
这里写图片描述
最后测试Python接口,这里建议如果不想把系统默认的Python版本设置为上面的版本的话,将上面添加到系统变量中的Python3.5的路径删除,之后再调用Caffe的pyhton接口时可以在Anaconda中调用,即首先打开Anaconda Prompt然后进行以下操作,如果没有报错,则说明Python的接口也配置成功。
这里写图片描述

  • 5
    点赞
  • 33
    收藏
    觉得还不错? 一键收藏
  • 16
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 16
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值