docker: Error response from daemon: Unknown runtime specified nvidia. See ‘docker run --help‘.报错的解决

在创建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