ubuntu16.04下caffe2与dectectron配置

一.caffe2配置

参考：Caffe2 - (一)Source 安装及问题解决

            caffe2官网

1.安装依赖项

sudo apt-get update
sudo apt-get install -y --no-install-recommends \
      build-essential \
      cmake \
      git \
      libgoogle-glog-dev \
      libgtest-dev \
      libiomp-dev \
      libleveldb-dev \
      liblmdb-dev \
      libopencv-dev \
      libopenmpi-dev \
      libsnappy-dev \
      libprotobuf-dev \
      openmpi-bin \
      openmpi-doc \
      protobuf-compiler \
      python-dev \
      python-pip                          
sudo pip install \
      future \
      numpy \
      protobuf

可能出现的问题：

Command "python setup.py egg_info" failed with error code 1 in /tmp/pip-build-t5Ju9r/future/

解决办法：升级setuptools

命令：pip install --upgrade setuptools

2.其他依赖项

sudo apt-get install -y --no-install-recommends libgflags-dev

根据博客Caffe2 - (一)Source 安装及问题解决的描述，还安装了其他的依赖库：

sudo pip install \
      flask \
      graphviz \
      hypothesis \
      jupyter \
      matplotlib \
      pydot python-nvd3 \
      pyyaml \
      requests \
      scikit-image \
      scipy \
      setuptools \
      tornado

其中pyyaml(3.13版本)、hypothesis、pydot、setuptools是必须要装的。（有些可能要利用命令sudo apt-get install，来安装）。

3.编译

# Clone Caffe2's source code from our Github repository
git clone --recursive https://github.com/pytorch/pytorch.git && cd pytorch
git submodule update --init --recursive

python setup.py install

这一步等的时间比较长

4.运行命令测试CPU-Caffe2是否编译成功

# To check if Caffe2 build was successful
python2 -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

# To check if Caffe2 GPU build was successful
# This must print a number > 0 in order to use Detectron
python2 -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'

如果显示以下则说明配置成功

5.运行命令测试 GPU-Caffe2是否编译成功

进入pytorch的目录，输入以下指令

python caffe2/python/operator_test/activation_ops_test.py

若出现，说明caffe2成功

可能出现的问题：

遇到这个问题，可以继续。

二.detectron配置

参考：点击打开链接   

           caffe2-detectron安装

1.COCO API安装

# COCOAPI=/home/yantianwang/clone/cocoapi
git clone https://github.com/cocodataset/cocoapi.git $COCOAPI
cd $COCOAPI/PythonAPI
# Install into global site-packages
make install
# Alternatively, if you do not have permissions or prefer
# not to install the COCO API into global site-packages
python2 setup.py install --user

可能出现的问题

1.执行到make install出现setup.py install running install error: can't create or remove files in install directory

解决办法：采用 sudo make install

或者是上述 python2 setup.py install --user

2.执行完毕后进入pycocotools应该有以下几个文件，当时我缺少了_mask.so

解决办法：不使用makefile，直接打开makefile，拷贝其中的指令，并改变为：

① python2 setup.py build_ext --inplace           

② python2 setup.py build_ext install

依次运行① ②，每一条运行完后删除build文件夹。一般情况下①指令运行会没有问题，②指令可能会出现权限问题，这时用python2 setup.py install --user 代替即可

2.下载detectron并编译

（1）clone detectron：

#DETECTRON=/home/yantianwang/clone/detectrongit clone https://github.com/facebookresearch/detectron $DETECTRON

（2）安装相关依赖项

pip install -r $DETECTRON/requirements.txt

可能会出现如下情况，不过没有影响

（3）编译

cd $DETECTRON && make

（4）添加环境变量

要将刚才我们编译的caffe2添加到环境变量中，在终端输入：

gedit ~/.bashrc

在其中添加：

# 随便写点啥表示你自己写的不是电脑加的，删除的时候好识别 
export PYTHONPATH=/home/yantianwang/clone/pytorch/build:$PYTHONPATH

之后再再终端输入：

source .bashrc

（5）检验

python2 $DETECTRON/detectron/tests/test_spatial_narrow_as_op.py

若出现以下则说明成功