1.4 Segformer
约 145 字小于 1 分钟
2025-04-20
SegFormer 要求 mmcv 版本在 [1.1.4,1.3.0]
之间,但是mmcv-full 这个版本之间不支持 pytorch>=1.9
,也就导致CUDA版本必须小于11.3
但是由于H100/H20的计算能力为90,只有11.8版本CUDA才能适配,所以用不了
解决:
使用最新的 mmsegmentation ,其整合了 SegFormer,并且适配 mmcv>=2.0.0
,也就适配我们的新版 pytorch 和 CUDA了
注意:
mmsegmentation 中的预训练模型与 SegFormer 其官方仓库的预训练模型不一样,需要重新使用 mmsegmentation 的链接下载