docker-compose.yml

version: "3"
services:
  app:
    image: oflow:latest
    build:
      context: .
      dockerfile: Dockerfile
    environment:
      - DISPLAY=${DISPLAY}
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=all
    volumes:
      - /tmp/.X11-unix:/tmp/.X11-unix
      - .:/work:rw
      - /etc/passwd:/etc/passwd:ro
      - /etc/group:/etc/group:ro
    user: 1000:1000
    network_mode: host
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]

  cuda11app:
    image: oflow:cuda11
    build:
      context: .
      dockerfile: Dockerfile_cuda11
    environment:
      - DISPLAY=${DISPLAY}
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=all
    volumes:
      - /tmp/.X11-unix:/tmp/.X11-unix
      - .:/work:rw
      - /etc/passwd:/etc/passwd:ro
      - /etc/group:/etc/group:ro
    user: 1000:1000
    network_mode: host
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]