目标检测小白 2022-10-25 12:58 采纳率: 25%
浏览 91
已结题

yolov5单机多卡训练报错

问题遇到的现象和发生背景

yolov5启动单机多卡训练报错

用代码块功能插入代码,请勿粘贴截图
运行结果及报错内容

img

  • 写回答

2条回答 默认 最新

  • 爱晚乏客游 2022-10-25 16:09
    关注

    详细说下你的配置,特别是系统,显卡型号和cuda,torch版本之类的,报错截图截取完整,比如执行到哪句代码的时候开始报错
    然后你的运行代码后面参数是什么,就是下面的语句后面的参数,如果batch size很大的话设置小一些看看
    python -m torch.distributed.run --nproc_per_node

    评论 编辑记录

报告相同问题?

问题事件

  • 已结题 (查看结题原因) 11月1日
  • 创建了问题 10月25日