yolov8多卡训练报错
时间: 2023-08-15 17:05:46 浏览: 402
对于yolov8多卡训练报错的问题,可以尝试以下解决方法。首先,确保使用的版本是正确的,因为有些版本的参数可能会有所不同。例如,在一些版本中,--save-period可能是--save_period。此外,确保batch-size能够被3整除,否则可能会导致错误。另外,可以尝试使用--nproc_per_node参数来指定使用的卡数,确保与实际使用的卡数一致。如果使用--nproc_per_node参数仍然出现问题,可以尝试使用以下启动命令:
python -m torch.distributed.launch --nproc_per_node=3 train.py --weights yolov5m.pt --img 640 --epoch 150 --data jm_fire_16.yaml --batch-size 36 --workers 16 --save-period 20。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [【深度学习 pytorch yolov 单机多卡的训练命令方式和注意事项](https://blog.csdn.net/weixin_40293999/article/details/127782051)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]
阅读全文