mmdetection环境配置和安装

1. 创建 openmmlab 虚拟环境

conda create -n openmmlab python=3.7
1

2. 激活openmmlab环境

conda activate openmmlab
1

3. 安装torch
   在torch官网查找安装自己电脑cuda对应的torch安装命令

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
1

4. 使用 MIM 安装 MMEngine 和 MMCV。

pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"
1
2
3

5. 安装 MMDetection

git clone https://github.com/open-mmlab/mmdetection.git
# 或 直接去官网下载压缩包
cd mmdetection   # 进入mmdetection目录
pip install -r requirements/build.txt
pip install -v -e .  #后面这个点不能少
# "-v" 指详细说明，或更多的输出
# "-e" 表示在可编辑模式下安装项目，因此对代码所做的任何本地修改都会生效，从而无需重新安装。
1
2
3
4
5
6
7

mmdetection官网使用指南