首先查看自己的cuda版本
nvcc --version
安装与cuda对应的torch版本,对应关系
Previous PyTorch Versions | PyTorch
安装对应的mmcv版本,使用下面命令可以自动安装匹配的版本,使用mim安装
mim install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cuda版本}/torch{torch版本号}/index.html
如果使用pip安装,我是报了这个错误
ModuleNotFoundError: No module named 'mmcv._ext'
最后安装mmcv对应的mmdet
依赖 — MMDetection 2.28.2 文档