File size: 1,358 Bytes
1d91be3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | 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"]
|