MMdetection香港中文大学MMLab最近开源了一个基于Pytorch实现的深度学习目标检测工具箱，支持Faster-RCNN，Mask-RCNN，Fast-RCNN等主流的目标检测框架，本文主要讲解MMdetection在滴滴云的机器学习工作室(DAI)下的安装方法。

滴滴云的机器学习工作室(DAI)，内置了丰富的机器学习开发环境，安装MMetection框架，总共需要3大步骤：

1、切换到所需环境，确认GCC环境

新建terminal，切换到pytorch环境:

conda info -e
conda activate ipykernel_py3_pytorch1.4

保证GCC>=5.0:

gcc -v

如果GCC<5.0:

sudo yum install centos-release-scl
sudo yum install devtoolset-7-gcc*
scl enable devtoolset-7 bash
gcc -v

2.创建环境变量

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
export CFLAGS="-I$CUDA_HOME/include $CFLAGS"

3.安装mmdetection：

pip install pycocotools
pip install mmcv
git clone https://github.com/open-mmlab/mmdetection
cd mmdetection
python setup.py develop