mmclassification 配置记录

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档

@TOCmmclassification 配置记录


前言

mmclassification出现
mmclassification 配置记录_第1张图片发现尤其是在使用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

你可能感兴趣的:(MMLab学习,python,人工智能,深度学习)