MMdetection(1)-环境安装

滴滴云技术支持发表于:2020年04月29日 14:55:13

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