使用docker里的环境运行程序,遇到错误:
引发错误的原因是共享内存不足。可通过如下命令查看容器中共享内存的大小:
df -h | grep shm
解决方法是新建一个容器,在 run 的时候添加参数--shm-size 8g
nvidia-docker run --shm-size 8g --name xinying-torch -it -v data/home/chenyu/xinying:/home/xinying 6a2d656bcf94 /bin/bash
另一个解决方法是进入宿主机中修改 Host.json,详参:如何修改容docker容器的shmsize共享内存大小。
【参考】