# GPU 叠加配置（Compose V2：docker compose）
# 用法：docker compose -f docker-compose.yml -f docker-compose.gpu-v2.yml up -d
#
# 前提：主机已安装 nvidia-container-toolkit
#   distribution=$(. /etc/os-release; echo $ID$VERSION_ID)
#   curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | sudo apt-key add -
#   curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list \
#     | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
#   sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
#   sudo nvidia-ctk runtime configure --runtime=docker
#   sudo systemctl restart docker

version: '3.8'

services:
  app:
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    environment:
      - ASSEMBLY_USE_GPU=1