RuntimeError: CUDA out of memory. Tried to allocate 52.00 MiB (GPU 0; 15.90 GiB total capacity; 14.85 GiB already allocated; 51.88 MiB free; 15.07 GiB reserved in total by PyTorch)
是显存炸掉了,可以改小batch_size,显存炸了以后,自然GPU也没有利用,因为模型尚未开始训练,所以显存利用率为0。
2G显存支持:SSD、YOLOV4-TINY
4G显存:YOLOV3
6G显存:YOLOV4、Retinanet、M2det、Efficientdet、Faster RCNN等
8G+显存均可。 需要注意的是,受到BatchNorm2d影响,batch_size不可为1,至少为2。