version: '2'

services:
  nvidia-ffmpeg:
    build: ./
    image: nvidia-ffmpeg
    container_name: nvidia-ffmpeg
    restart: "no"
    entrypoint:
      - /bin/sleep
      - infinity
    volumes:
      - /usr/lib/x86_64-linux-gnu/nvidia/current:/cuda:ro
      - ./folder:/folder
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
    devices:
      - /dev/nvidia0
      - /dev/nvidiactl
      - /dev/nvidia-uvm
      - /dev/nvidia-uvm-tools
    cap_add:
      - IPC_LOCK
    network_mode: host