根据官网介绍,caffe2安装可以选择源码编译安装和预编译安装两种方法。因为考虑到自己可能会对caffe2做改动,因此需要频繁编译安装;从而预编译安装的方式不符合这个需求,所以最后选择的是源码编译安装。
源码编译需要依赖许多第三方库,以下麻烦的地方:
root权限
第三方库容易不小心版本改变
易破坏别人的工作环境
这里决定使用anaconda的虚拟环境来编译安装caffe2。可以参考官网 Custom Anaconda Install
创建一个新的环境,比如我的命名为 caffe2-py27
conda create -n caffe2-py27 python=2.7
后面的python=2.7是说使用2.7的python版本
启动虚拟环境: source activate caffe2-py27
退出当前环境:source deactivate
一、安装
1. Create a conda environment
conda create -yn caffe2-py27 && source activate caffe2-py27
其中-y的意思是后面的问答环节都回答yes
caffe2-py27 是我自己的环境名字,可以随意替换。
2. Install required packages
conda install -y future gflags glog leveldb mkl mkl-include numpy opencv protobuf six
在安装mkl-include的时候,会有安装不上的现象,需要指定版本
image.png
换成以下命令,其他包按照原来安装. 其中anaconda可以替换成其他的(inte等,看官网有什么版本)
conda install -c anaconda mkl-include
3. Clone Caffe2's source code from our Github repository
cd ~ && git clone --recursive https://github.com/pytorch/pytorch.git && cd pytorch
git submodule update --init
这一步很慢,因为文件比较大
4. Create a directory to put Caffe2's build files in
rm -rf build && mkdir build && cd build
5. Configure Caffe2's build
This looks for packages on your machine and figures out which functionality to include in the Caffe2 installation. The output of this command is very useful in debugging.
cmake -DCMAKE_PREFIX_PATH=~/anaconda2/envs/caffe2-py27 -DCMAKE_INSTALL_PREFIX=~/anaconda2/envs/caffe2-py27 ..
6. Compile, link, and install Caffe2
make install