docker-compose.yaml

version: "3.9"
services:
  streamlit_app:
    build: .
    tty: true
    ports:
      - 8080:8080
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
  dev_app:
    image: tensorflow/tensorflow:latest-gpu
    tty: true
    volumes:
      - ./app:/app
      - ./root:/root
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]