docker-compose.yml

version: '3'

services:
  mongodb:
    build:
      context: apps
      dockerfile: mongo5.dockerfile
    network_mode: host
    volumes:
      - ./dataset/data:/data/db

  init_database:
    depends_on: [mongodb]
    build:
      context: apps
      dockerfile: Dockerfile
      args:
        APP_NAME: init_database
    command: ['python', 'src/init_database.py']
    volumes:
      - ./dataset:/dataset
    network_mode: host

  app:
    depends_on: [mongodb]
    build:
      context: apps
      dockerfile: Dockerfile
      args:
        APP_NAME: app
    network_mode: host

  notebooks:
    depends_on: [mongodb]
    build:
      context: apps
      dockerfile: Dockerfile_ubuntu
      args:
        APP_NAME: notebooks
    command: jupyter notebook --NotebookApp.token='' --allow-root --no-browser
    network_mode: host
    volumes:
      - ./dataset:/dataset
      - ./apps/notebooks/src:/app/notebooks/src
    devices:
      - /dev/nvidia0:/dev/nvidia0
      - /dev/nvidiactl:/dev/nvidiactl
      - /dev/nvidia-uvm:/dev/nvidia-uvm

  inference_app:
    build:
      context: apps
      dockerfile: Dockerfile_ubuntu
      args:
        APP_NAME: inference_service
    command: ['python3.9', 'src/inference_app.py', 'config/inference_config_docker.yml']
    network_mode: host
    volumes:
      - ./logs:/logs
      - ./models:/models
      - ./dataset:/dataset