最近想要学一下caffe,要想学首先得配置环境嘛,但是在第一步配置环境这里,就遇到了极大的阻碍。先不说caffe的版本有BLVC、Inter、windows等等,就在编译过程中遇到的各种bug,就让我不止一次的崩溃过。在调试了整整一周后某晚的一次尝试中立下誓言说编译过去就写博客分享,现在前来还愿,写完博客后我会重装一次系统再次重新编译尽量保证不会出错误。转载请注明出处。
首先放上参考过的前辈们的文章:
caffe-windows环境配置:https://www.cnblogs.com/jerrybaby/p/8385227.html
Windows10下安装py-faster-rcnn:https://blog.csdn.net/u010579901/article/details/78841857
如果你不需要Python接口的话可以看第一个链接,由于py-faster-rcnn的构建必须有Python接口,本问介绍有python接口的编译方式。
考虑到github上的代码会随着时间出现变化以及百度网盘链接失效导致本文不再适用的可能,我会把我编译成功版本的相关文件上传到csdn,以方便不喜欢折腾的小伙伴,因为安装环境时真的太!痛!苦!了!o(╥﹏╥)o
链接如下:
BVLC/caffe:https://download.csdn.net/download/u010853120/10601049
py-faster-rcnn和py-faster-rcnn-windows:https://download.csdn.net/download/u010853120/10601082
首先放上BLVC/caffe在的官方地址https://github.com/BVLC/caffe/tree/windows
1.安装vs2015,这里不用多说,大家自己百度下载,但是在安装时要选择自定义勾选上C++的语言编译器
2.安装cuda并配置cudnn
上英伟达的官网下载cuda,选择Legacy Releases,下载CUDA Toolkit 8.0 GA2。按你的操作系统和操作系统的版本选择exe(local)下载安装包。我的电脑配置是:
系统:windows10
cpu:i5-6400
gpu:gtx-950m
我的选择是
下载那个1.3G的的安装包,在下载的过程中我们去下载cudnn,申请个账号登录一下,进入cuDNN Download,选择 I Agree To the Terms of the cuDNN Software License Agreement后注意选择下面的Archived cuDNN Releases,下载cuDNN v5 (May 27, 2016), for CUDA 8.0。
安装cuda,一切选择都默认即可。等待安装完成后,把cudnn解压,把里面的每一个文件夹里面的文件copy到cuda安装目录下对应的文件夹内即可。如果不知道cuda安装在了哪里,可以通过此电脑-属性-高级系统设置-环境变量-系统变量里的CUDA_PATH查看。
3.安装anaconda的Python 2.7 version版本,默认选项即可
4.安装git,默认选项即可
5.打开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
7.下面进行caffe的编译
首先通过git下载源代码
git clone https://github.com/BVLC/caffe.git
cd caffe
git checkout windows
注意,重点来了,用vs2015打开scripts文件夹下的build_win.cmd并将第74行(不要改成文件开头的那个):
if NOT DEFINED WITH_NINJA set WITH_NINJA=1
修改为:
if NOT DEFINED WITH_NINJA set WITH_NINJA=0
再运行下面的代码:
scripts\build_win.cmd
这里会下载python的依赖包libraries_v140_x64_py27_1.1.0.tar.bz2,国内有时候需要才能下载
时间比较长,喝杯茶等一会出现下面的样子就好了,这里不知道为什么,如果我配置的没有问题编译的时候字体是像下面图片这样的黄色,如果有问题就会普通的cmd字体的颜色。
然后将“你存放caffe/python”的文件路径加入到python库中,代码如下
import sys
sys.path.insert(0,'你的文件路径/python')
import caffe
如果没有报错,那么恭喜你,caffe已经编译成功了
此处可以直接参考本文开头引用的第二个文章里面相关py-fasetr-rcnn的部分,但是该文章有两点我补充一下。
1.出现如下错误时,再cmd输入SET VS90COMNTOOLS=%VS120COMNTOOLS%后需要重启Anaconda Prompt才生效。
error: Microsoft Visual C++ 9.0 is required. Get it from http://aka.ms/vcpython27;
2.运行python ./tools/demo.py时发现缺少vgg16的.caffemodel文件,需要自己手动下载。由于文件过大,无法上传,所以需要大家自行百度。