使用环境:Pytorch1.8+CUDA11.1+RTX3060 Batch_Size = 4、Num_Worker = 8 报错CUDA OOM问题 解决办法: 1.Batch_Size设置为2 2.Num_Worker设置为4 补充链接: 1.https://www.zhihu.com/question/422160231 2.https://www.zhihu.com/question/400865628