docker 设置默认启动项

docker 设置默认启动项

问题说明

docker默认启动在cpu上，先要求调用宿主机nvidia显卡，配置默认runtime=nvidia

主要应用场景 在使用docker云边协同下发程序服务时，默认使用 nvidia gpu

实现功能

以下 docker run 等价

docker run -it image_name <=> docker run -it --gpus all image_name

或者

docker run -it image_name <=> docker run -it --runtime nvidia all image_name

# 安装nvidia-container-runtime，默认安装位置：/usr/bin/
sudo apt-get install nvidia-container-runtime
# 修改 /etc/docker/daemon.json，配置默认nvidia运行
sudo vim /etc/docker/daemon.json 

在daemon.json文件中添加如下内容，如下示意图

{
"default-runtime": "nvidia"
    "runtimes": {
        "nvidia": {
            "path": "/usr/bin/nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}

重启docker服务即可生效

sudo systemctl daemon-reload
sudo systemctl restart docker

参考引用文献

https://blog.csdn.net/u010420283/article/details/104055046