Docker容器中使用Pytorch训练时,当 DataLoader设置num_works>0时,训练程序崩溃,提示:
ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memory (shm)
解决办法如下:
在开启容器时使用如下命令 --shm-size 16G
Docker容器中使用Pytorch训练时,当 DataLoader设置num_works>0时,训练程序崩溃,提示:
ERROR: Unexpected bus error encountered in worker. This might be caused by insufficient shared memory (shm)
解决办法如下:
在开启容器时使用如下命令 --shm-size 16G