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的接口也配置成功。
更多推荐
所有评论(0)