根据官网介绍,caffe2安装可以选择源码编译安装和预编译安装两种方法。因为考虑到自己可能会对caffe2做改动,因此需要频繁编译安装;从而预编译安装的方式不符合这个需求,所以最后选择的是源码编译安装。
源码编译需要依赖许多第三方库,以下麻烦的地方:
- root权限
- 第三方库容易不小心版本改变
- 易破坏别人的工作环境
这里决定使用anaconda的虚拟环境来编译安装caffe2。可以参考官网 Custom Anaconda Install
零、Anaconda虚拟环境
- 创建一个新的环境,比如我的命名为 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的时候,会有安装不上的现象,需要指定版本
换成以下命令,其他包按照原来安装. 其中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
网友评论