Skip to main content
NVIDIA
PyTorch
Container
NVIDIA
PyTorch

PyTorch is a GPU accelerated tensor computational framework. Functionality can be extended with common Python libraries such as NumPy and SciPy. Automatic differentiation is done with a tape-based system at the functional and neural network layer levels.

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.ref=12711b2fffc2b532c812302ccd6b8c507ba1175d
    08/21/2026 6:45 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_REF=12711b2fffc2b532c812302ccd6b8c507ba1175d
    08/21/2026 6:45 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=406036884
    08/21/2026 6:45 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=406036884
    08/21/2026 6:45 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID=406036884
    08/21/2026 6:45 AM UTC
    sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1RUN
    NVIDIA_PYTORCH_VERSION=26.08 PYTORCH_BUILD_VERSION=2.14.0a0+4fdf77b NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+gitdd0efc75 TORCHTITAN_BUILD_VERSION=0.2.2+gitbadf21a1 TARGETARCH=amd64 PYVER=3.12 ENABLE_MITMPROXY=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali ENABLE_FIPS=0 /bin/bash -c if [ "${ENABLE_FIPS}" = "1" ] &&
      [ "${TARGETARCH}" = "amd64" ]; then echo "Running fips-fix.sh"; /tmp/fips-fix.sh; fi
    08/21/2026 6:45 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    ENABLE_FIPS=0
    08/21/2026 6:45 AM UTC
    sha256:a2f3b28823205c553a763e6ee506aaec6a8b66d53245708a4d29f7d0227eadffCOPY
    entrypoint.d/ /opt/nvidia/entrypoint.d/
    08/21/2026 6:45 AM UTC
    sha256:cc7d22b435f5184252bf88178fdc5c48931c12ba45b8a9ffd0d4878b9fee9177RUN
    NVIDIA_PYTORCH_VERSION=26.08 PYTORCH_BUILD_VERSION=2.14.0a0+4fdf77b NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+gitdd0efc75 TORCHTITAN_BUILD_VERSION=0.2.2+gitbadf21a1 TARGETARCH=amd64 PYVER=3.12 ENABLE_MITMPROXY=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali /bin/bash -c cp /tmp/restricted_constraint.txt /etc/pip/constraint.txt &&
      cp /tmp/original_constraint.txt /etc/pip/original_constraint.txt
    08/21/2026 6:45 AM UTC
    sha256:fb2c6da158871f4272bc7c7b10569bddb7df3eb7b8322638b968b3b8275ef44cRUN
    NVIDIA_PYTORCH_VERSION=26.08 PYTORCH_BUILD_VERSION=2.14.0a0+4fdf77b NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+gitdd0efc75 TORCHTITAN_BUILD_VERSION=0.2.2+gitbadf21a1 TARGETARCH=amd64 PYVER=3.12 ENABLE_MITMPROXY=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali /bin/bash -c ln -sf ${_CUDA_COMPAT_PATH}/lib.real ${_CUDA_COMPAT_PATH}/lib &&
      echo ${_CUDA_COMPAT_PATH}/lib > /etc/ld.so.conf.d/00-cuda-compat.conf &&
      ldconfig &&
      rm -f ${_CUDA_COMPAT_PATH}/lib
    08/21/2026 6:45 AM UTC
    ...