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