深度学习环境的一些坑(SegFormer)

wwwen 134 2022-07-29

我的环境介绍
RTX3060
UBUNTU22.04
cuda11.5
cudnn8.4.1 for cuda11.x

项目链接:SegFormer
RTX30系列不支持cuda10.x驱动
由于我最后才发现遇到了timm版本的问题,所以我不知道11.3版本cuda是否存在这个问题
但是mmseg最高支持到11,5我就用cuda11.5了
我的关键包版本分别是:
Pytorch = 1.11.0
Torchvision = 0.12.0
timm = 0.4.12
mvcc-full = 1.4.7

关键报错

ImportError: cannot import name 'container_abcs' from 'torch._six

wiki环境建议为timm==0.3.2报错
更新为0.4.12解决。

在Segformer的原代码中,如果你使用了更高版本的mvcc-full,那一定要去修改~/mmseg/__init__.py 中的MMCV-MAX参数,改为你的版本,这样就不会报错了。

Demo可视化

$ python demo/image_demo.py demo/demo.png local_configs/segformer/B0/segformer.b0.512x1024.city.160k.py checkpoint/segformer.b0.512x1024.city.160k.pth --device cuda:0 --palette cityscapes

SegFormerDEMO (1)