首先查看自己的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 文档

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐