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=0214a1e1471678f0c2928e94807c940c6961ca2c
    12/17/2025 7:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_REF=0214a1e1471678f0c2928e94807c940c6961ca2c
    12/17/2025 7:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
    com.nvidia.build.id=245654592
    12/17/2025 7:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    NVIDIA_BUILD_ID=245654592
    12/17/2025 7:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    NVIDIA_BUILD_ID=245654592
    12/17/2025 7:35 AM UTC
    sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1RUN
    NVIDIA_PYTORCH_VERSION=25.12 PYTORCH_BUILD_VERSION=2.10.0a0+b4e4ee8 NVFUSER_BUILD_VERSION=073e91b TORCHAO_BUILD_VERSION=+git01374eb5 TARGETARCH=arm64 PYVER=3.12 L4T=1 ENABLE_MITMPROXY=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali ENABLE_FIPS=0 if [ "${ENABLE_FIPS}" = "1" ] &&
      [ "${TARGETARCH}" = "amd64" ]; then echo "Running fips-fix.sh"; /tmp/fips-fix.sh; fi
    12/17/2025 7:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    TARGETARCH=arm64
    12/17/2025 7:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
    ENABLE_FIPS=0
    12/17/2025 7:35 AM UTC
    sha256:4315248befbbc189bee62f64bd059ee9454c3c678f0edc30e5a8a677f0310fc8COPY
    entrypoint.d/ /opt/nvidia/entrypoint.d/
    12/17/2025 7:35 AM UTC
    sha256:f06a1f99246a74cb50be489d31863796600870845dbc972ba5dbeb8b9d28475fCOPY
    constraint.txt /etc/pip/original_constraint.txt
    12/17/2025 7:35 AM UTC
    ...