24 lines
656 B
Text
24 lines
656 B
Text
# x86_64 CUDA 13 image for RunPod Blackwell distributed validation.
|
|
FROM pytorch/pytorch:2.9.1-cuda13.0-cudnn9-devel
|
|
|
|
ARG COMFY_KITCHEN_VERSION=0.2.31
|
|
|
|
RUN apt-get update \
|
|
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ffmpeg git \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /opt/h3-blackwell-runtime
|
|
COPY . .
|
|
|
|
RUN python -m pip install --no-cache-dir \
|
|
"comfy-kitchen==${COMFY_KITCHEN_VERSION}" \
|
|
"fastsafetensors>=0.1.10" \
|
|
"numpy>=2.0" \
|
|
"pillow>=11" \
|
|
"safetensors>=0.5.0" \
|
|
"scipy>=1.14" \
|
|
"transformers>=4.51,<5"
|
|
|
|
ENV PYTHONPATH=/opt/h3-blackwell-runtime/src
|
|
ENTRYPOINT []
|
|
CMD ["bash"]
|