mirror/docker.io/pytorch/pytorch (2.4.0-cuda12.1-cudnn9-runtime)

Published 2026-08-01 18:53:24 +00:00 by forgejo_admin

Installation

docker pull forge.adverant.ai/ci/mirror/docker.io/pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
sha256:68c022c2f4627943a6f3e574cfd2c8ae4256210d5f66ae2b117942e0a8d4fa9d

Image layers

ARG RELEASE
ARG LAUNCHPAD_BUILD_ARCH
LABEL org.opencontainers.image.ref.name=ubuntu
LABEL org.opencontainers.image.version=22.04
ADD file:d5da92199726e42da09a6f75a778befb607fe3f79e4afaf7ef5188329b26b386 in /
CMD ["/bin/bash"]
ARG PYTORCH_VERSION=2.4.0
ARG TRITON_VERSION=
ARG TARGETPLATFORM=linux/amd64
ARG CUDA_VERSION=12.1.1
LABEL com.nvidia.volumes.needed=nvidia_driver
RUN |4 PYTORCH_VERSION=2.4.0 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.1.1 /bin/sh -c apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends ca-certificates libjpeg-dev libpng-dev && rm -rf /var/lib/apt/lists/* # buildkit
COPY /opt/conda /opt/conda # buildkit
RUN |4 PYTORCH_VERSION=2.4.0 TRITON_VERSION= TARGETPLATFORM=linux/amd64 CUDA_VERSION=12.1.1 /bin/sh -c if test -n "${TRITON_VERSION}" -a "${TARGETPLATFORM}" != "linux/arm64"; then DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends gcc; rm -rf /var/lib/apt/lists/*; fi # buildkit
ENV PATH=/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
ENV NVIDIA_VISIBLE_DEVICES=all
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64
ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
ENV PYTORCH_VERSION=2.4.0
WORKDIR /workspace

Labels

Key Value
com.nvidia.volumes.needed nvidia_driver
org.opencontainers.image.ref.name ubuntu
org.opencontainers.image.version 22.04
Details
Container
2026-08-01 18:53:24 +00:00
1
OCI / Docker
linux/amd64
3.6 GiB
Versions (1) View all