YOLOv5使用GPU训练以后无法对目标图片进行识别

YOLO v5 使用GPU训练完成以后对目标图片进行识别显示以下代码，请问是哪里出问题了

oom错误，显存不足，将imgs这个参数改小，你这8032的的图片没几张显卡跑的了的

• 你可以参考下这个问题的回答, 看看是否对你有帮助, 链接: https://ask.csdn.net/questions/7427729
• 这篇博客你也可以参考下：3060显卡-yolov5-GPU环境搭建的一些建议
• 除此之外, 这篇博客: yolov5使用GPU中的 使用GPU 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  

• 

• 您还可以看一下 刘润森老师的yolov5目标检测教程课程中的 torch-gpu搭建小节, 巩固相关知识点