- 创建 openmmlab 虚拟环境
conda create -n openmmlab python=3.7
- 激活openmmlab环境
conda activate openmmlab
- 安装torch
在torch官网查找安装自己电脑cuda对应的torch安装命令
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
- 使用 MIM 安装 MMEngine 和 MMCV。
pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"
- 安装 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" 表示在可编辑模式下安装项目,因此对代码所做的任何本地修改都会生效,从而无需重新安装。
mmdetection官网使用指南