docker-compose.yml

version: '3.8'

services:
  deep_learning:
    container_name: ultra
    image: jhoeller/computer-vision:latest
    volumes:
      - ./app/:/app
    ports:
      - 8888:8888
    networks:
      - api_bridge
    deploy:
      resources:
        reservations:
          devices:
          - driver: nvidia
            count: 3  # num of gpus
            capabilities: [gpu]
    restart: always

  mlflow:
    image: jhoeller/mlflow-v.1.14.1:default
    ports:
      - 5000:5000
    volumes:
      - ./mlflow:/mlflow
    networks:
      - api_bridge
    restart: always
    command: mlflow server --backend-store-uri sqlite:////mlflow/mlflow.sqlite --default-artifact-root /mlflow/artifacts --host 0.0.0.0 

  tensorboard:
    image: tensorflow/tensorflow
    networks:
      - api_bridge
    volumes:
      - ./tensor:/notebooks
    ports:
      - 6006:6006
    command: tensorboard --logdir=/notebooks/graphs --bind_all

networks:
  api_bridge:
    driver: bridge

volumes:
  app_data:
  mlflow_data:
  tensor_data: