最近在使用YOLOv3进行测试时候,会报错cuda out of memory

最近在使用YOLOv3进行测试时候,会报错cuda out of memory,然后自己也尝试了一些方法,比如缩小测试图片尺寸等等,但是问题依然没有解决。想问一下各位大佬,有没有什么好的解决办法,或者YOLOv3有没有提供一些函数来释放显存,就是模型用完以后直接释放掉。谢谢谢!!!

https://blog.csdn.net/qq_27276951/article/details/102667894

仅限于pycharm调用darknet.py接口调用模型(如果你是的话,不然请忽略): 如果是2g及以上显存是不会有这个问题的,我查了一下yolov3模型大概1.6g的样子,你可能是多次加载了模型导致显存不足,你看一些darknet.py里你是不是每次测试图片时都调用了load_net这几个方法加载模型