Namespace
pytorch
Image / Tag
pytorch:2.5.0-cuda12.1-cudnn9-devel
Content Digest
sha256:9069f8ba74785313cebd5dda453ca9ae7fdb03d227143a687b3c6288853af4e7
Details
Created

2024-10-17 16:33:47 UTC

Size

6.59 GB

Content Digest
Labels
  • com.nvidia.volumes.needed
    nvidia_driver
  • maintainer
    NVIDIA CORPORATION <cudatools@nvidia.com>
  • org.opencontainers.image.ref.name
    ubuntu
  • org.opencontainers.image.version
    22.04

Environment
CUDA_VERSION

12.1.1

LD_LIBRARY_PATH

/usr/local/nvidia/lib:/usr/local/nvidia/lib64

LIBRARY_PATH

/usr/local/cuda/lib64/stubs

NCCL_VERSION

2.17.1-1

NVARCH

x86_64

NVIDIA_DRIVER_CAPABILITIES

compute,utility

NVIDIA_PRODUCT_NAME

CUDA

NVIDIA_REQUIRE_CUDA

cuda>=12.1 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=525,driver<526 brand=unknown,driver>=525,driver<526 brand=nvidia,driver>=525,driver<526 brand=nvidiartx,driver>=525,driver<526 brand=geforce,driver>=525,driver<526 brand=geforcertx,driver>=525,driver<526 brand=quadro,driver>=525,driver<526 brand=quadrortx,driver>=525,driver<526 brand=titan,driver>=525,driver<526 brand=titanrtx,driver>=525,driver<526

NVIDIA_VISIBLE_DEVICES

all

NV_CUDA_COMPAT_PACKAGE

cuda-compat-12-1

NV_CUDA_CUDART_DEV_VERSION

12.1.105-1

NV_CUDA_CUDART_VERSION

12.1.105-1

NV_CUDA_LIB_VERSION

12.1.1-1

NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE

cuda-nsight-compute-12-1=12.1.1-1

NV_CUDA_NSIGHT_COMPUTE_VERSION

12.1.1-1

NV_LIBCUBLAS_DEV_PACKAGE

libcublas-dev-12-1=12.1.3.1-1

NV_LIBCUBLAS_DEV_PACKAGE_NAME

libcublas-dev-12-1

NV_LIBCUBLAS_DEV_VERSION

12.1.3.1-1

NV_LIBCUBLAS_PACKAGE

libcublas-12-1=12.1.3.1-1

NV_LIBCUBLAS_PACKAGE_NAME

libcublas-12-1

NV_LIBCUBLAS_VERSION

12.1.3.1-1

NV_LIBCUSPARSE_DEV_VERSION

12.1.0.106-1

NV_LIBCUSPARSE_VERSION

12.1.0.106-1

NV_LIBNCCL_DEV_PACKAGE

libnccl-dev=2.17.1-1+cuda12.1

NV_LIBNCCL_DEV_PACKAGE_NAME

libnccl-dev

NV_LIBNCCL_DEV_PACKAGE_VERSION

2.17.1-1

NV_LIBNCCL_PACKAGE

libnccl2=2.17.1-1+cuda12.1

NV_LIBNCCL_PACKAGE_NAME

libnccl2

NV_LIBNCCL_PACKAGE_VERSION

2.17.1-1

NV_LIBNPP_DEV_PACKAGE

libnpp-dev-12-1=12.1.0.40-1

NV_LIBNPP_DEV_VERSION

12.1.0.40-1

NV_LIBNPP_PACKAGE

libnpp-12-1=12.1.0.40-1

NV_LIBNPP_VERSION

12.1.0.40-1

NV_NVML_DEV_VERSION

12.1.105-1

NV_NVPROF_DEV_PACKAGE

cuda-nvprof-12-1=12.1.105-1

NV_NVPROF_VERSION

12.1.105-1

NV_NVTX_VERSION

12.1.105-1

PATH

/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

PYTORCH_VERSION

2.5.0


Layers

[#000] sha256:43f89b94cd7df92a2f7e565b8fb1b7f502eff2cd225508cbd7ea2d36a9a3a601 - 0.43% (29 MB)

[#001] sha256:dd4939a047617694ff7585d2d828e574bcdb1a0e870d41ba1abad6db61fca814 - 0.07% (4.41 MB)

[#002] sha256:b0d7cc89b769c1d6a46d85393516b303afeeee8d8bcbcdadb9cc9a452728cf27 - 0.79% (53.1 MB)

[#003] sha256:1532d9024b9c102cb1b19cfd585e8735a1efc882bfa9243f6662b47e033c0c2c - 0.0% (185 Bytes)

[#004] sha256:04fc8a31fa53f7fd9ba45a2d109618449fd5f9d9b3f12d1cf961f4c0a0a34500 - 0.0% (6.72 KB)

[#005] sha256:a14a8a8a6ebc3813d37a448205bf2c059e7b0dde5dda741babfffc327f32638c - 18.77% (1.24 GB)

[#006] sha256:7d61afc7a3ac0a2b20643507f9ed50f42939d6838db55de1736527a711e98038 - 0.0% (62.4 KB)

[#007] sha256:8bd2762ffdd9e59311bc1d8909932d3363ca4897858f7f71aa837969b22754d8 - 0.0% (1.64 KB)

[#008] sha256:2a5ee6fadd42bc8a2088ecb52dc294274388aaa285d21507e399dbebfdb54f49 - 0.0% (1.49 KB)

[#009] sha256:22ba0fb08ae2d48e74c162458e138662946b32e12d711832122e465210ecbee8 - 35.98% (2.37 GB)

[#010] sha256:4d37a6bba88ff56fdbfcacd9ed2e4c2030b712adb162e0efa69b82ca7220a069 - 0.0% (86.3 KB)

[#011] sha256:537810544c714886df54bb5aa6f9432f4a5237955b2930371512d18510969c30 - 0.02% (1.44 MB)

[#012] sha256:f6d873ea5a6626cc92ee1626c56d8f72276a79885aeffe1eff42a6bd62efeac8 - 43.94% (2.9 GB)

[#013] sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1 - 0.0% (32 Bytes)

[#014] sha256:4774dabd8589dc3c932411b82c582a33b8c9abdf404ddc3d4f1b03aebe94c012 - 0.0% (99 Bytes)


History
2023-10-05 07:33:30 UTC

/bin/sh -c #(nop) ARG RELEASE

2023-10-05 07:33:30 UTC

/bin/sh -c #(nop) ARG LAUNCHPAD_BUILD_ARCH

2023-10-05 07:33:30 UTC

/bin/sh -c #(nop) LABEL org.opencontainers.image.ref.name=ubuntu

2023-10-05 07:33:30 UTC

/bin/sh -c #(nop) LABEL org.opencontainers.image.version=22.04

2023-10-05 07:33:32 UTC

/bin/sh -c #(nop) ADD file:63d5ab3ef0aab308c0e71cb67292c5467f60deafa9b0418cbb220affcd078444 in /

2023-10-05 07:33:32 UTC

/bin/sh -c #(nop) CMD ["/bin/bash"]

2023-11-10 05:07:58 UTC (buildkit.dockerfile.v0)

ENV NVARCH=x86_64

2023-11-10 05:07:58 UTC (buildkit.dockerfile.v0)

ENV NVIDIA_REQUIRE_CUDA=cuda>=12.1 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=525,driver<526 brand=unknown,driver>=525,driver<526 brand=nvidia,driver>=525,driver<526 brand=nvidiartx,driver>=525,driver<526 brand=geforce,driver>=525,driver<526 brand=geforcertx,driver>=525,driver<526 brand=quadro,driver>=525,driver<526 brand=quadrortx,driver>=525,driver<526 brand=titan,driver>=525,driver<526 brand=titanrtx,driver>=525,driver<526

2023-11-10 05:07:58 UTC (buildkit.dockerfile.v0)

ENV NV_CUDA_CUDART_VERSION=12.1.105-1

2023-11-10 05:07:58 UTC (buildkit.dockerfile.v0)

ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-12-1

2023-11-10 05:07:58 UTC (buildkit.dockerfile.v0)

ARG TARGETARCH

2023-11-10 05:07:58 UTC (buildkit.dockerfile.v0)

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

2023-11-10 05:07:58 UTC (buildkit.dockerfile.v0)

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.0-1_all.deb && dpkg -i cuda-keyring_1.0-1_all.deb && apt-get purge --autoremove -y curl && rm -rf /var/lib/apt/lists/* # buildkit

2023-11-10 05:07:58 UTC (buildkit.dockerfile.v0)

ENV CUDA_VERSION=12.1.1

2023-11-10 05:08:11 UTC (buildkit.dockerfile.v0)

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-12-1=${NV_CUDA_CUDART_VERSION} ${NV_CUDA_COMPAT_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit

2023-11-10 05:08:12 UTC (buildkit.dockerfile.v0)

RUN |1 TARGETARCH=amd64 /bin/sh -c echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf # buildkit

2023-11-10 05:08:12 UTC (buildkit.dockerfile.v0)

ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

2023-11-10 05:08:12 UTC (buildkit.dockerfile.v0)

ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64

2023-11-10 05:08:12 UTC (buildkit.dockerfile.v0)

COPY NGC-DL-CONTAINER-LICENSE / # buildkit

2023-11-10 05:08:12 UTC (buildkit.dockerfile.v0)

ENV NVIDIA_VISIBLE_DEVICES=all

2023-11-10 05:08:12 UTC (buildkit.dockerfile.v0)

ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_CUDA_LIB_VERSION=12.1.1-1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_NVTX_VERSION=12.1.105-1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_LIBNPP_VERSION=12.1.0.40-1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_LIBNPP_PACKAGE=libnpp-12-1=12.1.0.40-1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_LIBCUSPARSE_VERSION=12.1.0.106-1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-12-1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_LIBCUBLAS_VERSION=12.1.3.1-1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_LIBCUBLAS_PACKAGE=libcublas-12-1=12.1.3.1-1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_LIBNCCL_PACKAGE_VERSION=2.17.1-1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NCCL_VERSION=2.17.1-1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NV_LIBNCCL_PACKAGE=libnccl2=2.17.1-1+cuda12.1

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ARG TARGETARCH

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-libraries-12-1=${NV_CUDA_LIB_VERSION} ${NV_LIBNPP_PACKAGE} cuda-nvtx-12-1=${NV_NVTX_VERSION} libcusparse-12-1=${NV_LIBCUSPARSE_VERSION} ${NV_LIBCUBLAS_PACKAGE} ${NV_LIBNCCL_PACKAGE} && rm -rf /var/lib/apt/lists/* # buildkit

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-mark hold ${NV_LIBCUBLAS_PACKAGE_NAME} ${NV_LIBNCCL_PACKAGE_NAME} # buildkit

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

COPY entrypoint.d/ /opt/nvidia/entrypoint.d/ # buildkit

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

COPY nvidia_entrypoint.sh /opt/nvidia/ # buildkit

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENV NVIDIA_PRODUCT_NAME=CUDA

2023-11-10 05:13:35 UTC (buildkit.dockerfile.v0)

ENTRYPOINT ["/opt/nvidia/nvidia_entrypoint.sh"]

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_CUDA_LIB_VERSION=12.1.1-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_CUDA_CUDART_DEV_VERSION=12.1.105-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_NVML_DEV_VERSION=12.1.105-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_LIBCUSPARSE_DEV_VERSION=12.1.0.106-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_LIBNPP_DEV_VERSION=12.1.0.40-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-12-1=12.1.0.40-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_LIBCUBLAS_DEV_VERSION=12.1.3.1-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME=libcublas-dev-12-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_LIBCUBLAS_DEV_PACKAGE=libcublas-dev-12-1=12.1.3.1-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.1.1-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-1=12.1.1-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_NVPROF_VERSION=12.1.105-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_NVPROF_DEV_PACKAGE=cuda-nvprof-12-1=12.1.105-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.17.1-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NCCL_VERSION=2.17.1-1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.17.1-1+cuda12.1

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ARG TARGETARCH

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

LABEL maintainer=NVIDIA CORPORATION <cudatools@nvidia.com>

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-get update && apt-get install -y --no-install-recommends cuda-cudart-dev-12-1=${NV_CUDA_CUDART_DEV_VERSION} cuda-command-line-tools-12-1=${NV_CUDA_LIB_VERSION} cuda-minimal-build-12-1=${NV_CUDA_LIB_VERSION} cuda-libraries-dev-12-1=${NV_CUDA_LIB_VERSION} cuda-nvml-dev-12-1=${NV_NVML_DEV_VERSION} ${NV_NVPROF_DEV_PACKAGE} ${NV_LIBNPP_DEV_PACKAGE} libcusparse-dev-12-1=${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

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

RUN |1 TARGETARCH=amd64 /bin/sh -c apt-mark hold ${NV_LIBCUBLAS_DEV_PACKAGE_NAME} ${NV_LIBNCCL_DEV_PACKAGE_NAME} # buildkit

2023-11-10 05:25:51 UTC (buildkit.dockerfile.v0)

ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs

2024-10-17 16:30:05 UTC (buildkit.dockerfile.v0)

ARG PYTORCH_VERSION=2.5.0

2024-10-17 16:30:05 UTC (buildkit.dockerfile.v0)

ARG TRITON_VERSION=

2024-10-17 16:30:05 UTC (buildkit.dockerfile.v0)

ARG TARGETPLATFORM=linux/amd64

2024-10-17 16:30:05 UTC (buildkit.dockerfile.v0)

ARG CUDA_VERSION=12.1.1

2024-10-17 16:30:05 UTC (buildkit.dockerfile.v0)

LABEL com.nvidia.volumes.needed=nvidia_driver

2024-10-17 16:30:05 UTC (buildkit.dockerfile.v0)

RUN |4 PYTORCH_VERSION=2.5.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

2024-10-17 16:33:46 UTC (buildkit.dockerfile.v0)

COPY /opt/conda /opt/conda # buildkit

2024-10-17 16:33:47 UTC (buildkit.dockerfile.v0)

RUN |4 PYTORCH_VERSION=2.5.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

2024-10-17 16:33:47 UTC (buildkit.dockerfile.v0)

ENV PATH=/opt/conda/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

2024-10-17 16:33:47 UTC (buildkit.dockerfile.v0)

ENV NVIDIA_VISIBLE_DEVICES=all

2024-10-17 16:33:47 UTC (buildkit.dockerfile.v0)

ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility

2024-10-17 16:33:47 UTC (buildkit.dockerfile.v0)

ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64

2024-10-17 16:33:47 UTC (buildkit.dockerfile.v0)

ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin

2024-10-17 16:33:47 UTC (buildkit.dockerfile.v0)

ENV PYTORCH_VERSION=2.5.0

2024-10-17 16:33:47 UTC (buildkit.dockerfile.v0)

WORKDIR /workspace

Danger Zone
Delete Tag

Please be careful as this will not just delete the reference but also the actual content!

For example when you have latest and v1.2.3 both pointing to the same image
the deletion of latest will also permanently remove v1.2.3.

Delete