| FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04 | |
| ENV DEBIAN_FRONTEND=noninteractive | |
| # Python + dependencies | |
| RUN apt-get update && apt-get install -y python3 python3-pip git && \ | |
| pip3 install --upgrade pip | |
| # Set working dir | |
| WORKDIR /app | |
| # Copy and install requirements | |
| COPY requirements.txt ./ | |
| RUN pip install --no-cache-dir -r requirements.txt | |
| # Copy app code | |
| COPY . . | |
| CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "5000"] | |