提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
@TOCmmclassification 配置记录
前言
mmclassification出现
发现尤其是在使用vit相关backbone时候会出现这个问题。
经过反复实验,最终合理的配置如下:
1,CUDA11.3 最新的cudnn11.x
2,torch1.10 torchvision0.11.1 torchaudio0.10.0
https://download.pytorch.org/whl/torch_stable.html
3,MMCV1.7.0
链接: https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html