nvidia-docker-compose.yml

nvidia-docker-compose.yml

version: '2.0'
services:
  kirin-yolo-filter:
    image: kirin-yolo-filter:latest
    build:
      context: .
      dockerfile: Dockerfile
    volumes:
      - nvidia_driver_384.111:/usr/local/nvidia:ro
    devices:
      - /dev/nvidia0:/dev/nvidia0
      - /dev/nvidiactl:/dev/nvidiactl
      - /dev/nvidia-uvm:/dev/nvidia-uvm
    ports:
      - 8989:8989
    command: bash -c "python3 -u app.py --port=8989"
    networks:
      - mlflow
networks:
  mlflow:
    external: true

volumes:
  nvidia_driver_384.111:
    external: true