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