官网:https://github.com/BVLC/caffe/tree/windows
一、前期准备
Requirements
Visual Studio 2013 or 2015
Technically only the VS C/C++ compiler is required (cl.exe)
CMake3.4 or higher (Visual Studio andNinjagenerators are supported)
电脑上已经安装visual studio 2013,但是缺少CMake,故在cmd中用pip install cmake安装。
二、配置编译
C:\Projects>git clone https://github.com/BVLC/caffe.git
C:\Projects>cdcaffeC:\Projects\caffe>git checkout windows
::Edit any of the options inside build_win.cmd to suit your needs
C:\Projects\caffe>scripts\build_win.cmd
a.随便选个位置存放caffe项目
b.在运行build_win.cmd这个命令之前需要编辑build_win.cmd文件
这里可以根据自己的配置修改,我使用的vs2013故将MSVC_VERSION的值设为12,不使用Ninja,故WITH_NINJA=0。(其他配置类似,根据需求修改即可)
c.配置完之后,网络稳定可直接用scripts\build_win.cmd进行后续操作。
这里由于我的网络不稳定,无法下载libraries_v120_x64_py27_1.1.0.tar.bz2,故先搜索该安装包下载(该包百度云链接:https://pan.baidu.com/s/1jHDg24u 密码:1wgd)
下载后将该压缩包放到C:\Users\Administrator\.caffe\dependencies\download路径下之后使用scripts\build_win.cmd即可进行编译,编译过程大概五分钟完成。(有573个警告,不知 道什么意思,暂时忽略,反正没报错)
这时,在.\caffe\build\tools\Release目录下已经成功生成caffe.exe啦~
将caffe\python下的caffe文件夹复制到Anaconda\Lib\site-packages下即可。
三、测试
import caffe
from caffe import layers as L
from caffe import params as P
输入以上内容,运行测试,我这里报了错,提示ImportError: No module named google.protobuf.internal,解决办法很简单,缺什么补什么嘛~
直接conda install protobuf,安装完后再测试一下,没有报错就可以啦