caffe和caffe 2都是优秀的深度学习框架。
自从2013年,贾杨清在2013.9.13发布了caffe的第一个版本,caffe就受到了全世界开发者的青睐,时至今日,caffe在github在已经收获了超过22000颗星,并在工业界广泛使用,成为最受欢迎的深度学习框架之一。
2017年4月,caffe 2发布,官方介绍说:这是一个轻量化和模块化的深度学习框架,在强调轻便性的同时,也保持了可扩展性和计算性能。Caffe2擅长移动和大规模部署,更偏向于工业应用,而且重点关注在移动端上的表现。
caffe和caffe2都是值得深入学习的框架。目前caffe和caffe2都已经比较成熟,相比于刚发布时繁杂的安装步骤,当前caffe和caffe2的安装已经简单了不少,具体的安装步骤,官方都给出了比较详细的说明,但在安装的过程中还是遇到了不少的问题。本文旨在记录初学者在win10系统下安装caffe和caffe2遇到的各种问题,大神勿喷。
安装环境
windows 10, visual studio 2015 (目前caffe支持vs2013和2015,caffe2支持vs2015和vs2017, 为了支持二者,选择了vs2015)
依赖软件
Anaconda Python 2.7 or 3.5 x64 (or Miniconda)
Anaconda是一个用于科学计算的Python发行版,包含了NumPy,SciPy等众多科学计算库,Miniconda是Anoconda的简化版,可以根据自己的需求安装相应的库,可以避免冗余库的安装,节省空间。
目前caffe支持python2.7和3.5,而caffe2支持python2.7, 为了支持二者,本文选择安装Anaconda Python2.7.
cuda 8.0
cuDNN v5
caffe安装步骤
首先,按照官网步骤来,
C:\Projects> git clone https://github.com/BVLC/caffe.git
C:\Projects> cd caffe
C:\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
结果出现下面的错误:
Snipaste_2018-01-12_22-07-07.png
不用急,打开scripts文件夹下的build_win.cmd,修改WITH_NINJA = 0,CPU_ONLY = 1,(这里只编译CPU版本,因为cuda编译很慢,而且学习代码,暂时没有必要编译GPU版本),同时删除build文件夹下的CMakeFile和CMakeCash.txt.
Snipaste_2018-01-12_22-10-51.png
再次运行scripts/build_wind.cmd, 静静地等待caffe配置完成。另外,如果网速不太好的话,可能会无法完成依赖包的下载,默认情况下caffe的依赖包下载在C:\Users\用户名.caffe\dependencies\download的路径下,如果无法下载下来,可以到网上直接找别人已经下好的包,比如libraries_v140_x64_py27_1.1.0.tar.bz2。
caffe 2安装步骤
有了安装caffe的基础,caffe2的安装就简单多了,下载代码
git clone --recursive https://github.com/caffe2/caffe2.git
然后在scripts目录下的build_windows.bat和build_host_protoc.bat添加
set CMAKE_GENERATOR="Visual Studio 14 2015 Win64"
指定编译器。
Snipaste_2018-01-12_22-31-02.png
最后运行,运行scripts/build_windows.bat,坐等config完成,就是这么简单。(PS:注意caffe 2下载的目录文件夹名称如果是空格,中间不可以有空格,否则会有问题)