MMDetection 环境配置

一、环境配置步骤

1. 创建虚拟环境

conda create -n open-mmlab python=3.7 -y
conda activate open-mmlab

创建虚拟环境名称open-mmlab，python版本选择3.7（和官方文档一致）

2. 安装torch、torchvision

conda install pytorch=1.6.0 torchvision=0.7.0 -c pytorch

注意：确保CUDA版本和torch的匹配，我这里CUDA版本为10.2。

3. 安装mmcv（版本需要和torch吻合）

官方文档中下载mmcv-full推荐格式如下：

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html

 我这里的CUDA版本为10.2，torch版本为1.6.0，则格式如下：

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.6.0/index.html

4. 下载mmdetection

这里有个小trick，可以加快git clone速度（加中国域名）：

git clone https://github.com.cnpmjs.org/open-mmlab/mmdetection.git

5. 安装编译mmdetection

cd mmdetection
pip install -r requirements/build.txt
pip install -v -e . # or “python setup.py develop”

6. 测试环境是否可用

python tools/train.py -h

 出现如下结果，表示安装成功：

 二、参考链接

1. https://blog.csdn.net/weixin_37707770/article/details/113775866

2. https://github.com/open-mmlab/mmdetection/blob/master/docs/get_started.md