| ARG RELEASE |
| ARG LAUNCHPAD_BUILD_ARCH |
| LABEL org.opencontainers.image.ref.name=ubuntu |
| LABEL org.opencontainers.image.version=22.04 |
| ADD file:1b6c8c9518be42fa2afe5e241ca31677fce58d27cdfa88baa91a65a259be3637 in / |
| CMD ["/bin/bash"] |
| ENV NVARCH=x86_64 |
| ENV NVIDIA_REQUIRE_CUDA=cuda>=12.8 brand=unknown,driver>=470,driver<471 brand=grid,driver>=470,driver<471 brand=tesla,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=vapps,driver>=470,driver<471 brand=vpc,driver>=470,driver<471 brand=vcs,driver>=470,driver<471 brand=vws,driver>=470,driver<471 brand=cloudgaming,driver>=470,driver<471 brand=unknown,driver>=535,driver<536 brand=grid,driver>=535,driver<536 brand=tesla,driver>=535,driver<536 brand=nvidia,driver>=535,driver<536 brand=quadro,driver>=535,driver<536 brand=quadrortx,driver>=535,driver<536 brand=nvidiartx,driver>=535,driver<536 brand=vapps,driver>=535,driver<536 brand=vpc,driver>=535,driver<536 brand=vcs,driver>=535,driver<536 brand=vws,driver>=535,driver<536 brand=cloudgaming,driver>=535,driver<536 brand=unknown,driver>=550,driver<551 brand=grid,driver>=550,driver<551 brand=tesla,driver>=550,driver<551 brand=nvidia,driver>=550,driver<551 brand=quadro,driver>=550,driver<551 brand=quadrortx,driver>=550,driver<551 brand=nvidiartx,driver>=550,driver<551 brand=vapps,driver>=550,driver<551 brand=vpc,driver>=550,driver<551 brand=vcs,driver>=550,driver<551 brand=vws,driver>=550,driver<551 brand=cloudgaming,driver>=550,driver<551 brand=unknown,driver>=560,driver<561 brand=grid,driver>=560,driver<561 brand=tesla,driver>=560,driver<561 brand=nvidia,driver>=560,driver<561 brand=quadro,driver>=560,driver<561 brand=quadrortx,driver>=560,driver<561 brand=nvidiartx,driver>=560,driver<561 brand=vapps,driver>=560,driver<561 brand=vpc,driver>=560,driver<561 brand=vcs,driver>=560,driver<561 brand=vws,driver>=560,driver<561 brand=cloudgaming,driver>=560,driver<561 brand=unknown,driver>=565,driver<566 brand=grid,driver>=565,driver<566 brand=tesla,driver>=565,driver<566 brand=nvidia,driver>=565,driver<566 brand=quadro,driver>=565,driver<566 brand=quadrortx,driver>=565,driver<566 brand=nvidiartx,driver>=565,driver<566 brand=vapps,driver>=565,driver<566 brand=vpc,driver>=565,driver<566 brand=vcs,driver>=565,driver<566 brand=vws,driver>=565,driver<566 brand=cloudgaming,driver>=565,driver<566 |
| ENV NV_CUDA_CUDART_VERSION=12.8.90-1 |
| ARG TARGETARCH |
| LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com> |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends gnupg2 curl ca-certificates && curl -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${NVARCH}/cuda-keyring_1.1-1_all.deb && dpkg -i cuda-keyring_1.1-1_all.deb && apt-get purge --autoremove -y curl && rm -rf /var/lib/apt/lists/* # buildkit |
| ENV CUDA_VERSION=12.8.1 |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-12-8=${NV_CUDA_CUDART_VERSION} cuda-compat-12-8 && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |1 TARGETARCH=amd64 /bin/sh -c echo "/usr/local/cuda/lib64" >> /etc/ld.so.conf.d/nvidia.conf # buildkit |
| ENV PATH=/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64 |
| COPY NGC-DL-CONTAINER-LICENSE / # buildkit |
| ENV NVIDIA_VISIBLE_DEVICES=all |
| ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility |
| ENV NV_CUDA_LIB_VERSION=12.8.1-1 |
| ENV NV_NVTX_VERSION=12.8.90-1 |
| ENV NV_LIBNPP_VERSION=12.3.3.100-1 |
| ENV NV_LIBNPP_PACKAGE=libnpp-12-8=12.3.3.100-1 |
| ENV NV_LIBCUSPARSE_VERSION=12.5.8.93-1 |
| ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-12-8 |
| ENV NV_LIBCUBLAS_VERSION=12.8.4.1-1 |
| ENV NV_LIBCUBLAS_PACKAGE=libcublas-12-8=12.8.4.1-1 |
| ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2 |
| ENV NV_LIBNCCL_PACKAGE_VERSION=2.25.1-1 |
| ENV NCCL_VERSION=2.25.1-1 |
| ENV NV_LIBNCCL_PACKAGE=libnccl2=2.25.1-1+cuda12.8 |
| ARG TARGETARCH |
| LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com> |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-libraries-12-8=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-12-8=${NV_NVTX_VERSION} libcusparse-12-8=${NV_LIBCUSPARSE_VERSION} ${NV_LIBCUBLAS_PACKAGE} ${NV_LIBNCCL_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-mark hold ${NV_LIBCUBLAS_PACKAGE_NAME} ${NV_LIBNCCL_PACKAGE_NAME} # buildkit |
| COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit |
| COPY nvidia_entrypoint.sh /opt/nvidia/ # buildkit |
| ENV NVIDIA_PRODUCT_NAME=CUDA |
| ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"] |
| ENV NV_CUDA_LIB_VERSION=12.8.1-1 |
| ENV NV_CUDA_CUDART_DEV_VERSION=12.8.90-1 |
| ENV NV_NVML_DEV_VERSION=12.8.90-1 |
| ENV NV_LIBCUSPARSE_DEV_VERSION=12.5.8.93-1 |
| ENV NV_LIBNPP_DEV_VERSION=12.3.3.100-1 |
| ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-12-8=12.3.3.100-1 |
| ENV NV_LIBCUBLAS_DEV_VERSION=12.8.4.1-1 |
| ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME=libcublas-dev-12-8 |
| ENV NV_LIBCUBLAS_DEV_PACKAGE=libcublas-dev-12-8=12.8.4.1-1 |
| ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.8.1-1 |
| ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-8=12.8.1-1 |
| ENV NV_NVPROF_VERSION=12.8.90-1 |
| ENV NV_NVPROF_DEV_PACKAGE=cuda-nvprof-12-8=12.8.90-1 |
| ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev |
| ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.25.1-1 |
| ENV NCCL_VERSION=2.25.1-1 |
| ENV NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.25.1-1+cuda12.8 |
| ARG TARGETARCH |
| LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com> |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-dev-12-8=${NV_CUDA_CUDART_DEV_VERSION} cuda-command-line-tools-12-8=${NV_CUDA_LIB_VERSION} cuda-minimal-build-12-8=${NV_CUDA_LIB_VERSION} cuda-libraries-dev-12-8=${NV_CUDA_LIB_VERSION} cuda-nvml-dev-12-8=${NV_NVML_DEV_VERSION} ${NV_NVPROF_DEV_PACKAGE} ${NV_LIBNPP_DEV_PACKAGE} libcusparse-dev-12-8=${NV_LIBCUSPARSE_DEV_VERSION} ${NV_LIBCUBLAS_DEV_PACKAGE} ${NV_LIBNCCL_DEV_PACKAGE} ${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN |1 TARGETARCH=amd64 /bin/sh -c apt-mark hold ${NV_LIBCUBLAS_DEV_PACKAGE_NAME} ${NV_LIBNCCL_DEV_PACKAGE_NAME} # buildkit |
| ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs |
| ENV DEBIAN_FRONTEND=noninteractive |
| ENV PYTHONUNBUFFERED=1 |
| ENV PIP_NO_CACHE_DIR=1 |
| ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64 |
| ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs:/usr/local/cuda/lib64/stubs |
| ENV PATH=/usr/local/cuda/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin |
| ENV HF_HOME=/workspace/.cache/huggingface |
| ENV HF_HUB_ENABLE_HF_TRANSFER=1 |
| ENV TRANSFORMERS_CACHE=/workspace/.cache/huggingface/transformers |
| RUN /bin/sh -c apt-get update && apt-get install -y software-properties-common && add-apt-repository -y ppa:deadsnakes/ppa && apt-get update && apt-get install -y python3.12 python3.12-dev python3.12-venv python3-pip git git-lfs wget curl ffmpeg libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender-dev libgomp1 build-essential ninja-build && rm -rf /var/lib/apt/lists/* # buildkit |
| RUN /bin/sh -c update-alternatives --install /usr/bin/python python /usr/bin/python3.12 1 && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.12 1 && python3.12 -m ensurepip --upgrade && python3.12 -m pip install --upgrade pip setuptools wheel # buildkit |
| RUN /bin/sh -c pip install torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128 triton==3.4.0 --index-url https://download.pytorch.org/whl/cu128 && rm -rf /root/.cache/pip /tmp/* # buildkit |
| RUN /bin/sh -c pip install https://github.com/nunchaku-tech/nunchaku/releases/download/v1.0.2/nunchaku-1.0.2+torch2.8-cp312-cp312-linux_x86_64.whl # buildkit |
| COPY requirements.txt /tmp/requirements.txt # buildkit |
| RUN /bin/sh -c pip install -r /tmp/requirements.txt && rm -rf /root/.cache/pip # buildkit |
| WORKDIR /tmp |
| ENV EXT_PARALLEL=2 |
| ENV NVCC_APPEND_FLAGS=--threads 2 |
| ENV MAX_JOBS=4 |
| ENV TORCH_CUDA_ARCH_LIST=9.0 |
| RUN /bin/sh -c git clone https://github.com/thu-ml/SageAttention.git && cd SageAttention && pip install --no-build-isolation . && cd / && rm -rf /tmp/SageAttention /root/.cache/pip # buildkit |
| WORKDIR /workspace |
| RUN /bin/sh -c git clone https://github.com/comfyanonymous/ComfyUI.git && cd ComfyUI && pip install -r requirements.txt # buildkit |
| WORKDIR /workspace/ComfyUI/custom_nodes |
| RUN /bin/sh -c git clone https://github.com/ltdrdata/ComfyUI-Manager.git && git clone https://github.com/jnxmx/ComfyUI_HuggingFace_Downloader.git && git clone https://github.com/kijai/ComfyUI-KJNodes.git && git clone https://github.com/Fannovel16/comfyui_controlnet_aux.git && git clone https://github.com/crystian/ComfyUI-Crystools.git && git clone https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite.git && git clone https://github.com/willmiao/ComfyUI-Lora-Manager.git && git clone https://github.com/city96/ComfyUI-GGUF.git && git clone https://github.com/Fannovel16/ComfyUI-Frame-Interpolation.git && git clone https://github.com/nunchaku-tech/ComfyUI-nunchaku.git && git clone https://github.com/evanspearman/ComfyMath.git && git clone https://github.com/ssitu/ComfyUI_UltimateSDUpscale.git # buildkit |
| RUN /bin/sh -c (cd ComfyUI-KJNodes && pip install -r requirements.txt || true) && (cd comfyui_controlnet_aux && pip install -r requirements.txt || true) && (cd ComfyUI-VideoHelperSuite && pip install -r requirements.txt || true) && (cd ComfyUI-GGUF && pip install -r requirements.txt || true) && (cd ComfyUI-Frame-Interpolation && pip install -r requirements.txt || true) && (cd ComfyUI-nunchaku && pip install -r requirements.txt || true) && rm -rf /root/.cache/pip /tmp/* # buildkit |
| WORKDIR /workspace/ComfyUI |
| RUN /bin/sh -c mkdir -p /runpod-volume/models/checkpoints /runpod-volume/models/loras /runpod-volume/models/vae /runpod-volume/models/controlnet /runpod-volume/models/clip /runpod-volume/models/upscale_models /runpod-volume/.cache/huggingface /workspace/.cache # buildkit |
| RUN /bin/sh -c rm -rf models/checkpoints && ln -s /runpod-volume/models/checkpoints models/checkpoints && rm -rf models/loras && ln -s /runpod-volume/models/loras models/loras && rm -rf models/vae && ln -s /runpod-volume/models/vae models/vae && rm -rf models/controlnet && ln -s /runpod-volume/models/controlnet models/controlnet && rm -rf models/clip && ln -s /runpod-volume/models/clip models/clip && rm -rf models/upscale_models && ln -s /runpod-volume/models/upscale_models models/upscale_models # buildkit |
| RUN /bin/sh -c ln -s /runpod-volume/.cache/huggingface /workspace/.cache/huggingface # buildkit |
| WORKDIR /workspace |
| COPY handler.py /workspace/handler.py # buildkit |
| COPY workflows /workspace/workflows # buildkit |
| CMD ["python" "-u" "handler.py"] |
