一、下载源码
二、准备工作
注意:我们的源码是0.6版本,那么mmdet、mmengine、mmcv必须按照下图要求安装,记得检查
mmcv的安装不是那么简单,首先根据源码版本,确定mmcv=2.0.0;然后选择系统、cuda版本、torch版本,得到mmcv的安装命令行。
注意:mmcv版本确定了,那么torch版本不能过高,否则报错,安装torch时格外注意!
详见:Installation — mmcv 2.1.0 documentation
接前言,安装torch不要直接在这个表里选,点击进入之前版本,选择正确版本安装:
三、开始部署
创建并激活一个 conda 环境:
conda create --name mmyolo python=3.10
conda activate mmyolo
安装pytorch(命令上面有说):
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
验证 PyTorch 安装:
安装mim:
pip install -U openmim
根据mmyolo源码下的requirements/mminstall.txt进行安装:
mim install "mmengine>=0.7.1"
mim install "mmdet>=3.0.0"
mmcv的安装命令行如下(上面有说): pip install mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0/index.html
mim install -r requirements/albu.txt
mim install -v -e .