在创建nvidia/cuda构建的docker镜像的时候, 报错
其中create_container.sh脚本里代码如下:
#!/bin/bash docker run -itd -m 32G --memory-swap -1 -p 10013:22 -v /files/:/files/ --runtime=nvidia --name nvidia_dev01 nvidia_dev bin/bash -c "service ssh start && /bin/bash"
首先确保nvidia-docker有安装
sudo apt install nvidia-docker2
然后/etc/docker/daemon.json里的内容如下:
{ "registry-mirrors": ["https://f1z25q5p.mirror.aliyuncs.com"], "runtimes": { "nvidia": { "path": "nvidia-container-runtime", "runtimeArgs": [] } } }
然后执行如下两条命令即可
sudo systemctl daemon-reload
sudo systemctl restart docker