| FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime | |
| WORKDIR /app | |
| # System deps for OpenCV | |
| RUN apt-get update && apt-get install -y --no-install-recommends \ | |
| libgl1-mesa-glx libglib2.0-0 ffmpeg curl && \ | |
| rm -rf /var/lib/apt/lists/* | |
| COPY requirements.txt . | |
| RUN pip install --no-cache-dir -r requirements.txt opencv-python-headless>=4.8.0 | |
| COPY model/ model/ | |
| COPY agents/ agents/ | |
| COPY api/ api/ | |
| COPY edge/ edge/ | |
| COPY data/ data/ | |
| COPY configs/ configs/ | |
| COPY evaluation/ evaluation/ | |
| COPY autokernel/ autokernel/ | |
| COPY train.py . | |
| EXPOSE 8000 | |
| CMD ["uvicorn", "api.main:app", "--host", "0.0.0.0", "--port", "8000", "--workers", "1"] | |