yolov5单机多卡训练报错

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

yolov5启动单机多卡训练报错

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

img

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