这个屌丝很懒，什么也没留下！

YOLOv8多卡训练报错TypeError: barrier() got an unexpected keyword argument ‘device_ids‘

作者：我家小花儿 | 2024-06-13 02:45:02

训练yolov8时，device指定多GPU训练报错（单卡能跑）：

yolo task=detect mode=train model=yolov8n.yaml device=0,1,2,3

原因：pytorch版本问题，为了适配边缘计算盒子，用的是1.7版本的pytorch

解决：升级pytorch版本，1.9以后的大概就可以，官方推荐torch>=2.0.0版本，pytorch升级命令可查看Previous PyTorch Versions | PyTorch

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/710586