MMdetection(1)-环境安装

滴滴云技术支持发表于:2020年04月29日 14:34:03更新于:2020年04月29日 14:56:39

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

在滴滴云的GPU实例安装MMetection框架,总共需要3大步骤:

1.需保证基本环境满足:

  • python=3.7

  • cuda=10.1.105(Linux)

  • cudnn=7.6.1.34(CUDA10.1) 

cuda,cudnn快速安装方法:https://help.didiyun.com/hc/kb/article/1344589/

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 torch==1.4.0 torchvison
pip install pycocotools
pip install mmcv
git clone https://github.com/open-mmlab/mmdetection
cd mmdetection
python setup.py develop