version: '3'

services:
  nvidia-ffmpeg:
    build: ./
    image: nvidia-ffmpeg
    container_name: nvidia-ffmpeg
    restart: "no"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              capabilities: [gpu]
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
    entrypoint:
      - /bin/sleep
      - infinity
    volumes:
      - ./folder:/folder
      - /usr/lib/x86_64-linux-gnu/nvidia/current:/cuda:ro
    network_mode: host