| # Use official PyTorch image with CUDA 12.1 runtime | |
| FROM pytorch/pytorch:2.1.2-cuda12.1-cudnn8-runtime | |
| WORKDIR /app | |
| # Install system dependencies including OpenMP for multi-threaded GBDTs | |
| RUN apt-get update && apt-get install -y --no-install-recommends \ | |
| build-essential \ | |
| libgomp1 \ | |
| curl \ | |
| git \ | |
| && rm -rf /var/lib/apt/lists/* | |
| # Install Python requirements | |
| COPY requirements_hf.txt . | |
| RUN pip install --no-cache-dir -r requirements_hf.txt | |
| # Copy application and benchmark runner scripts | |
| COPY run_hf_pro_high_budget.py . | |
| COPY app.py . | |
| EXPOSE 7860 | |
| CMD ["python", "app.py"] | |