docker-compose.cuda.yml

version: "3"
services:
  postgres:
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [ gpu ]
    healthcheck:
      test: [ "CMD-SHELL", "pg_isready" ]
      interval: 1s
      timeout: 5s
      retries: 100
    build:
      context: ./pgml-extension/
      dockerfile: Dockerfile
    ports:
      - "5433:5432"
    command:
      - sleep
      - infinity
    volumes:
      - postgres-data:/var/lib/postgresql/14/main/

  dashboard:
    depends_on:
       postgres:
         condition: service_healthy
    build:
      context: ./pgml-dashboard/
      dockerfile: Dockerfile
    ports:
      - "8000:8000"
    environment:
      ROCKET_ADDRESS: 0.0.0.0
      DATABASE_URL: postgres://postgres:postgres@postgres:5432/pgml_development
    command: bash -c "sqlx migrate run && cargo run"
  docs:
    build:
      context: ./pgml-docs/
      dockerfile: Dockerfile
    ports:
      - "8001:8001"
    command:
      - mkdocs
      - serve
      - -a 0.0.0.0:8001
volumes:
  postgres-data: