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=9fa5c48351cf93ac6e6972ca113a7e3c54675a76`	01/22/2026 9:54 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`NVIDIA_BUILD_REF=9fa5c48351cf93ac6e6972ca113a7e3c54675a76`	01/22/2026 9:54 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	LABEL	`com.nvidia.build.id=256811083`	01/22/2026 9:54 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ENV	`NVIDIA_BUILD_ID=256811083`	01/22/2026 9:54 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`NVIDIA_BUILD_ID=256811083`	01/22/2026 9:54 AM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1	RUN	`NVIDIA_PYTORCH_VERSION=26.01 PYTORCH_BUILD_VERSION=2.10.0a0+a36e1d3 NVFUSER_BUILD_VERSION=5d8efce TORCHAO_BUILD_VERSION=+git1272f3cf TORCHTITAN_BUILD_VERSION=0.2.0+gite98ae995 TARGETARCH=amd64 PYVER=3.12 L4T=0 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`	01/22/2026 9:54 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`TARGETARCH=amd64`	01/22/2026 9:54 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`ENABLE_FIPS=0`	01/22/2026 9:54 AM UTC
sha256:b566d86978fee5b21af97fdc49078c6c5baa96fcb401821b8895be3b64705f05	COPY	`entrypoint.d/ /opt/nvidia/entrypoint.d/`	01/22/2026 9:54 AM UTC
sha256:dbd8305a00b27c73fe0671b227fc1bff530fc686ad91b77ef949f0a31dea8b64	COPY	`constraint.txt /etc/pip/original_constraint.txt`	01/22/2026 9:54 AM UTC