YOLOv8 GPU训练启动停滞无进展问题求助

强制关闭Mosaic数据增强 #

你的CPU训练流程会自动关闭Mosaic，而GPU环境下可能因该增强模块的兼容性问题导致停滞，直接在训练时禁用：

• 命令行启动：yolo detect train data=你的数据集配置.yaml model=yolov8n.pt epochs=50 mosaic=0
• Python脚本启动：在训练参数中指定mosaic=0，示例代码：

from ultralytics import YOLO

model = YOLO('yolov8n.pt')
model.train(data='你的数据集配置.yaml', epochs=50, mosaic=0)

降低DataLoader工作线程数 #

当前设置的8个workers可能超出GTX 1080（8G显存）的IO或显存承载能力，尝试调低线程数：

• 命令行添加参数：workers=4或workers=2
• Python脚本修改：model.train(..., workers=4)

排查显存占用情况 #

Mosaic增强需要加载多张图片拼接，可能导致显存耗尽后静默停滞：

• 训练前执行nvidia-smi查看初始显存占用
• 启动训练后持续运行nvidia-smi，观察显存是否被占满后停止变化

降级Ultralytics版本 #

你使用的v8.0.54版本可能存在GPU环境下Mosaic的兼容性bug，尝试切换到更稳定的旧版本：

• pip方式：pip install ultralytics==8.0.20
• conda方式：先卸载现有包，再安装指定版本

内容的提问来源于stack exchange，提问作者John Prescott