win10+visual studio 2015安装caffe和caffe 2

  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)

依赖软件

  • cmake 最新版3.10,下载地址https://cmake.org/files/v3.10/cmake-3.10.1-win64-x64.msi
  • 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.
  • Matlab 2016b
    链接:http://pan.baidu.com/s/1bpw6HDP 密码:uaja
  • 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下载的目录文件夹名称如果是空格,中间不可以有空格,否则会有问题)

参考文献
https://www.cnblogs.com/CQBZOIer-zyy/p/5933954.html

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容