yyxz18281455491的博客

原创 torch.cuda.OutOfMemoryError: CUDA out of memory.

使用YOLOv5跑模型的时候，遇到了torch.cuda.OutOfMemoryError: CUDA out of memory.的问题。使用了其他办法依然爆显存，然后我将img-size从640改到了256，就ok了。爆显存的问题，如果硬件支持的话，可以加个SSD，然后将SSD硬盘当虚拟内存。我的显卡是3060，显存6g，使用nvidia-smi命令查看显存充足。爆显存问题暂时解决，但是缩小图片尺寸可能会影响检测精度。