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
2c6982d19ea147bdaaceb69d1e22cad31e0aee741a04a78eecba0f07266d9717	CONFIG	`Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp`	01/10/2026 1:20 AM UTC
2a4e17988238a28bc41f47f11447c64855f05fccde3dee72c492f3c15f455761	ARG	`NVIDIA_BUILD_REF=0a3698acc72a53dab4a78b9e9504baa4a0e292d4`	01/10/2026 1:20 AM UTC
c25474a423b51e8ccaff5d2f554c07231a2785091c4c8ce388f654d56fa4c20e	LABEL	`com.nvidia.build.id=252182147`	01/10/2026 1:20 AM UTC
211c14481eb1b07663df6317b23b05d6145935cce5232de6f9b5478b4ebf9fc9	ENV	`NVIDIA_BUILD_ID=252182147`	01/10/2026 1:20 AM UTC
6468fdea925e3bd6838642fc5c9f8f500f76739be21cc15b654f6cce41dfbd6d	ARG	`NVIDIA_BUILD_ID=252182147`	01/10/2026 1:20 AM UTC
f05521a0d232fdc9f117d2705f897add539d56f6d5b2301d80a05d1f990eaff5	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=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`	01/10/2026 1:20 AM UTC
8b1ceb8130549a0171ee6db8b56d464177f8bf5d9e34b0ae8dfd1799058b6c40	ARG	`TARGETARCH=arm64`	01/10/2026 1:20 AM UTC
47fae30212c269e810c2ac99878cac60316d0fd438dae5366cb4124d2d8e8be5	ARG	`ENABLE_FIPS=0`	01/10/2026 1:20 AM UTC
035145867a4829cb5b6f823eb493f44ee5429a4f26df3c50d952164211cd9635	COPY	`entrypoint.d/ /opt/nvidia/entrypoint.d/`	01/10/2026 1:20 AM UTC
7c27fcf078c3479611105c6733531c4ee459b6b1366d44d7d281f83cf5354ab8	COPY	`constraint.txt /etc/pip/original_constraint.txt`	01/10/2026 1:20 AM UTC