version: "3.3"

services:

  vpf:
    build:
      context: ../
      dockerfile: docker/Dockerfile.gpu
      args:
        GEN_PYTORCH_EXT: 0
        GEN_OPENGL_EXT: 0
    ipc: host
    runtime: nvidia
    image: nvidia/videoprocessingframework:vpf
    environment:
      - NVIDIA_DRIVER_CAPABILITIES=video,compute,utility

  vpf-tensorrt:
    build:
      context: ../
      dockerfile: docker/Dockerfile-tensorrt.gpu
      args:
        GEN_PYTORCH_EXT: 0
        GEN_OPENGL_EXT: 0
    ipc: host
    runtime: nvidia
    image: nvidia/videoprocessingframework:vpf-tensorrt
    environment:
      - NVIDIA_DRIVER_CAPABILITIES=video,compute,utility