Simple Faster R-CNN运行代码出现cuda runtime error(2):out of memory at..解决方案_simple-faster-rcnn runtimeerror: cuda out of memor

在使用github上的simple faster rcnn-pytorch跑代码的时候，运行到一半会异常中止
出现如图所示的错误：
cuda runtime error(2):out of memory at…
但通过查询显卡情况发现显存明显够用

尝试了多种解决方案如：调整batch_size，调用torch.cuda.empty_cache()等方式都无法解决该问题。

    test_dataloader = data_.DataLoader(testset,
                                       batch_size=1,
                                       num_workers=opt.test_num_workers,
                                       shuffle=False, \
                                       pin_memory=False
                                       )

1
2
3
4
5
6
7

最终回到问题出现的pin_memory这一关键点，通过将dataloader里的pin_memory置为False即可正常运行。