前言
这个系列是记录我个人在学习过程中,遇到的一些问题,并记录我成功的方法,真实有效。
BEVFormer本身的安装文档非常不利好于新人,所以在此记录了我遇到的坑与解决办法,帮助和我一样初学入门BEV的朋友们。
使用平台
i7-12700k
4070ti
ubuntu20.04
conda虚拟环境中,python=3.8
遇到的问题
使用官方的安装文档,运行如下训练软件代码时,会报错:RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
运行此处的代码时,如果你是个人学生党,使用自己的电脑,只有一张显卡,请将结尾的8改为1。
以下是解决办法
这个错误表示你的GPU显存爆了。
官方提供了很多版本的训练:
他们从左到右分别需要的GPU显存以此减小,small > 10G,tiny > 6.7G,根据你的显卡能力选择适合你的,所以我选择了small,更改官方代码为:
./tools/dist_train.sh ./projects/configs/bevformer/bevformer_samll.py 8
即可成功解决