| FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu20.04 | |
| # System setup | |
| RUN apt-get update && apt-get install -y python3-pip git libgl1 | |
| # Python dependencies | |
| COPY requirements.txt . | |
| RUN pip3 install --upgrade pip && pip3 install -r requirements.txt | |
| # App | |
| COPY . /app | |
| WORKDIR /app | |
| CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] | |