yolov8 多卡训练报错subprocess.CalledProcessError: Command‘[‘/home/... returned non-zero exit status 1._yolov8多卡训练报错

作者：2023面试高手 | 2024-06-13 02:45:28

踩

yolov8多卡训练报错

在用yolov8多个卡进行训练时报以下错误
在这里插入图片描述
解决方法：
参考网上教程，在DistributedDataParallel 中加入

find_unused_parameters=True
1

加入位置为：ultralytics-main/ultralytics/engine/trainer.py 中的第247行
在这里插入图片描述

 self.model = nn.parallel.DistributedDataParallel(self.model, device_ids=[RANK], find_unused_parameters=True)
1

之后用命令行的方式重新运行训练起来了

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