报错内容:

RuntimeError: DataLoader worker (pid 186829) 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.

原因:
DataLoader的worker由于共享内存不足而被系统终止。PyTorch的DataLoader使用了多进程(num_workers > 0)来加速数据加载,而每个工作进程都需要一定的共享内存。如果系统的共享内存不足,就会触发 Bus error 错误。

解决方法:

1、临时增加系统共享内存

sudo mount -o remount,size=8G /dev/shm

2、永久增加

sudo nano /etc/fstab

添加或修改:
tmpfs /dev/shm tmpfs defaults,size=8G 0 0

重启系统

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐