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