FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04

ENV DEBIAN_FRONTEND=noninteractive \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=1

RUN apt-get update && apt-get install -y --no-install-recommends \
    build-essential \
    ca-certificates \
    curl \
    git \
    wget \
    libgl1 \
    libglib2.0-0 \
    libsm6 \
    libxext6 \
    libxrender1 \
    && rm -rf /var/lib/apt/lists/*

RUN wget -q https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /tmp/miniconda.sh \
    && bash /tmp/miniconda.sh -b -p /opt/conda \
    && rm /tmp/miniconda.sh

ENV PATH=/opt/conda/bin:$PATH

RUN conda create -y -n skillzero python=3.12 && conda clean -afy
SHELL ["conda", "run", "-n", "skillzero", "/bin/bash", "-c"]

WORKDIR /workspace/SkillZero

# Copy the repository into the image, or bind-mount it at runtime.
# Example build context should be the SkillZero repo root:
#   docker build -f export_packages/skillzero_best_checkpoints_20260506/Dockerfile -t skillzero:export .
COPY . /workspace/SkillZero

RUN pip install -U pip setuptools wheel \
    && pip install -r requirements.txt \
    && pip install -e . \
    && pip install flash-attn==2.7.4.post1 --no-build-isolation

ENV HF_HOME=/workspace/hf \
    HUGGINGFACE_HUB_CACHE=/workspace/hf/hub \
    HYDRA_FULL_ERROR=1

CMD ["/bin/bash"]