torch.cuda.OutOfMemoryError: CUDA out of memory.解决方案（训练yolov7时遇到的问题）

在训练yolov7时遇到了这个问题：

然后命令行输入nvidia-smi查看发现显存明明还有很多，所以网上很多杀进程的方法不是很奏效；

然后就一直看报错信息，发现这一行“batch_norm”，然后突发奇想，难道是batch太大了？同时放不进去？

接着把train.py文件中的batch从16改为8，代码运行成功，问题果然解决，我是12GB的显存，如果你们的显存更小的话可以改为4、2都行。