PyTorch | NVIDIA NGC

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.

NVIDIA AI Enterprise Supported

Layer	Label		Created
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	LABEL	`com.nvidia.build.ref=19e454c67f233643cfde9928b142a3e2173292bc`	09/23/2026 6:51 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`NVIDIA_BUILD_REF=19e454c67f233643cfde9928b142a3e2173292bc`	09/23/2026 6:51 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	LABEL	`com.nvidia.build.id=453215200`	09/23/2026 6:51 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ENV	`NVIDIA_BUILD_ID=453215200`	09/23/2026 6:51 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`NVIDIA_BUILD_ID=453215200`	09/23/2026 6:51 PM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1	RUN	`NVIDIA_PYTORCH_VERSION=26.09 PYTORCH_BUILD_VERSION=2.14.0a0+b2c75dd NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+git4ee58b57 TORCHTITAN_BUILD_VERSION=0.2.2+git5c0b804c 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`	09/23/2026 6:51 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`ENABLE_FIPS=0`	09/23/2026 6:51 PM UTC
sha256:8551ac1ccc9aefd620f50fe141e2d18d99dd08ef97c8fb6b79ffd55363a56c3f	COPY	`entrypoint.d/ /opt/nvidia/entrypoint.d/`	09/23/2026 6:51 PM UTC
sha256:0f62a44e1a8688063d5a300585a0309ddd11e2b28e929ef143da473d569c2e89	RUN	`NVIDIA_PYTORCH_VERSION=26.09 PYTORCH_BUILD_VERSION=2.14.0a0+b2c75dd NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+git4ee58b57 TORCHTITAN_BUILD_VERSION=0.2.2+git5c0b804c 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`	09/23/2026 6:51 PM UTC
sha256:da38876ae8d15039437da4fbf73a96145bcc40e72f566f0dcd16b81e04ad892b	RUN	`NVIDIA_PYTORCH_VERSION=26.09 PYTORCH_BUILD_VERSION=2.14.0a0+b2c75dd NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+git4ee58b57 TORCHTITAN_BUILD_VERSION=0.2.2+git5c0b804c 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`	09/23/2026 6:51 PM UTC