关于RuntimeError: CUDA out of memory. (OOM)尽管GPU利用率为0
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、YOLO















