cuda.docker-compose.yml

version: "3.9"
services:
  streamlit:
    container_name: streamlit
    build:
      context: ./app
      dockerfile: cuda.Dockerfile
    depends_on:
      - postgres
    restart: on-failure
    volumes:
      - ./app/.streamlit:/root/.streamlit
      - ./app/tmp:/app/tmp
      - ./streamlit-volume:/streamlit-volume
    env_file:
      - .env
    environment:
      - POSTGRES_USER=user
      - POSTGRES_PASSWORD=pass
      - POSTGRES_DB=postgres
      - POSTGRES_HOST=postgres
      - POSTGRES_PORT=5432
    ports:
      - ${SERVER_ADDR}:${HOST_PORT}:8501
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
  postgres:
    container_name: postgresql
    build: 
      context: ./postgresql
      dockerfile: Dockerfile
    restart: on-failure
    ports:
      - "127.0.0.1:5432:5432"
    environment:
      - POSTGRES_USER=user
      - POSTGRES_PASSWORD=pass
      - POSTGRES_DB=postgres
    volumes:
      - ./postgresql/data:/var/lib/postgresql/data