caffe2安装

根据官网介绍,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的时候,会有安装不上的现象,需要指定版本
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

你可能感兴趣的:(caffe2安装)