cuda12.1安装mmcv（AssertionError: MMCV==2.2.0 is used but incompatible. Please install mmcv＞=2.0.0rc4.）

刚开始尝试了很多方法，比如下载mmcv-full，但是这个库最高更新到1.7.2，下载之后显示mmcv的版本变成了1.7.2.最终原因大概是因为在 MMCV2.x 中，将 mmcv-full 重命名为 mmcv，将 mmcv 重名为 mmcv-lite。还有在GitHub上下载mmcv的安装包，手动安装mmcv，一样更新到了最高版mmcv2.2.0，报错+n。包冲突真的令人头大，还好最后解决了，标

m0_49201875

602人浏览 · 2024-11-06 09:47:33

m0_49201875 · 2024-11-06 09:47:33 发布

一、问题：

AssertionError: MMCV==2.2.0 is used but incompatible. Please install mmcv>=2.0.0rc4.

二、踩坑

刚开始尝试了很多方法，比如下载mmcv-full，但是这个库最高更新到1.7.2，下载之后显示mmcv的版本变成了1.7.2.最终原因大概是因为在 MMCV2.x 中，将 mmcv-full 重命名为 mmcv，将 mmcv 重名为 mmcv-lite。所以好像是一样的？

还有在GitHub上下载mmcv的安装包，手动安装mmcv，一样更新到了最高版mmcv2.2.0，报错+n

直接pip install mmcv=2.0.0rc4，还是报错......

三、解决

主要是mmcv版本不匹配

查看cuda版本号：nvcc -V(V要大写）（踩坑：刚开始使用nvidia -smi查看的，结果显示cuda12.4，但是mmcv官网并没有cuda12.4的匹配版本，于是开始了install 和 uninstall的循环...）
torch版本在requirements.txt文件中查看，我的是torch2.1.2
mmcv官网的匹配文件：https://mmcv.readthedocs.io/en/latest/get_started/installation.html
复制下面的pip 指令，我的是

pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu121/torch2.1/index.html

包冲突真的令人头大，还好最后解决了，标记一下（手动撒花*★,°*:.☆(￣▽￣)/$:*.°★* 。）

Triton中文社区

欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

torch.device(‘cuda:0‘ if cuda else ‘cpu‘)

# 这个地方的cuda:0 实际上并不是0号GPU，他取决于CUDA_VISIBLE_DEVICES# 然后逻辑GPU和物理GPU有一个对应关系# 如果CUDA_VISIBLE_DEVICES为2,1,3# 那么CUDA:0就是2号GPU， CUDA:1 就是1号GPU CUDA:3 就是3号GPUreturn torch.device('cuda:0' if cuda else 'cpu')..