docker 超出共享内存限制 RuntimeError: DataLoader

前言

docker 运行时报错：

RuntimeError: DataLoader worker (pid 86) is killed by signal: Bus error. It is possible that dataloader's workers are out of shared memory. Please try to raise your shared memory limit.

解决

调整 Docker 容器的共享内存限制

通过设置 Docker 容器的 --shm-size 参数来增加共享内存的大小。

例如，要分配 2GB 的共享内存，在运行 Docker 容器时使用以下命令

docker run --shm-size=2g -it your_image_name

docker run --shm-size 2G -it your_image_name