Ubuntu 18.04 docker-compose方式使用gpu

环境：
Ubuntu 18.04
Docker version 19.03.14
docker-compose version 1.24.1
NVIDIA Driver Version 460.39

1）安装docker-compose

2） 安装nvidia-container-runtime

sudo apt install nvidia-container-runtime

3） 修改配置文件/etc/docker/daemon.json使docker-compose支持nvidia-container-runtime

{
    "default-runtime":"nvidia",
    "runtimes": {
        "nvidia": {
            "path": "nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}
sudo systemctl daemon-reload
sudo systemctl restart docker

4） 修改docker-compose.yml
version 改为2.3
runtime 设置 nvidia
environment 设置 NVIDIA_VISIBLE_DEVICES: all
eg:

version: '2.3'
services:
    xxx:
        build: .
        container_name: xxx
        image: xxx
        runtime: nvidia
        environment:
            NVIDIA_VISIBLE_DEVICES: all

注意：

• 基础镜像使用ubuntu 18.04即可，不需要使用带gpu的docker image。