ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memory (shm)
简介:
使用YOLO11 在Docker 里面训练,出现一堆报错, 下面是关键报错信息:
ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memory (shm)
RuntimeError: DataLoader worker (pid 838) is killed by signal: Bus error. It is possible that dataloader's workers are out of shared memory.
代码原文:
1 2 3 4 5 6 7 8 | # ImageWoof 数据集 分析狗 from ultralytics import YOLO # Load a model model = YOLO( "yolo11n-cls.pt" ) # load a pretrained model (recommended for training) # Train the model model.train(data = "imagewoof160" , epochs = 100 , imgsz = 224 , workers = 4 ) |
问题分析:
翻译得知明显是共享内存不足(shm)问题,原因是YOLO 使用PyTorch 训练,PyTorch 使用共享内存在进程之间共享数据,当workers 比较多的时候容器运行时的默认共享内存段大小是不够的,比如workers 改为2就不报错了。
GitHub 官方原文: https://github.com/pytorch/pytorch#docker-image
解决方法:
1、docker run 时使用--ipc=host或--shm-size命令行选项增加共享内存大小
docker run -itd --name=yolo --shm-size=2g
# 查看共享内存使用率
docker exec yolo df -h /dev/shm
2、将 workers 改的小点。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具