仓库引用:mmdetection仓库
最近需要复现一些传统算法,所以需要借助mmdetection仓库来复现类似faster-rcnn等benchmark。这里就记录一下该架构的Linux版本环境搭建。
1、设备情况
我的设备情况如下:


如果跟我环境类似,可以仿照这套流程进行环境搭建。
2、环境搭建
按照以下步骤来创建环境:
bash
git clone https://github.com/Yhc-777/mmdetection.git
conda create -n mmdetection python=3.10 -y
conda activate mmdetection
cd mmdetection
# cuda安装参考:https://pytorch.org/get-started/previous-versions/
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
# 安装依赖
# 使用 pip 安装 openmin
pip install -U openmim
mim install mmcv==2.0.1
mim install mmdet==3.0.0
mim install mmrotate==1.0.0rc1
3、环境验证
终端输入:
bash
# 进入交互环境
python
import mmdet
print(mmdet.__version__)
显示:

即可。