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
8b961697c09affc35d953438cb49939b1849bfbf370bfa1807fcc7e194798443	CONFIG	`Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp`	01/22/2026 9:54 AM UTC
f8ff743f624790ff2de796fca5d0c8b1798dfb0484734637297e4bb5a566df06	ARG	`NVIDIA_BUILD_REF=9fa5c48351cf93ac6e6972ca113a7e3c54675a76`	01/22/2026 9:54 AM UTC
ad16c5c5c60996a1e3b5310eff7617f95dee280c16e459e0cc6bd8457121fa31	LABEL	`com.nvidia.build.id=256811083`	01/22/2026 9:54 AM UTC
514a1e62d8d230a46e47b1bf970baa93656110c8df9faa6b3208fb3f9ad53a04	ENV	`NVIDIA_BUILD_ID=256811083`	01/22/2026 9:54 AM UTC
236ebf6db28ae686b12410a89eb6da51a1690a9482f47c1a69d33205b4d25f45	ARG	`NVIDIA_BUILD_ID=256811083`	01/22/2026 9:54 AM UTC
125f35a8e08c19b238a3041d38d6338b4c3040c5aa7aa9b51257d0285799c985	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
a33f9f7b092d37d08c9e6cc7b9d6bcead8d36028472920709f1a9b52acb3a7f2	ARG	`TARGETARCH=amd64`	01/22/2026 9:54 AM UTC
808cf846356d18480e265819993bf66ade032112e9b77b5cbc37892f039e5dd0	ARG	`ENABLE_FIPS=0`	01/22/2026 9:54 AM UTC
992e887172874943a4729641630746c2243a8a1b7d6bb433c5376f0149f83a4d	COPY	`entrypoint.d/ /opt/nvidia/entrypoint.d/`	01/22/2026 9:54 AM UTC
6bc91df8e5b859b26f602e46b7cde81f756451b648213f5609b2401f8d108945	COPY	`constraint.txt /etc/pip/original_constraint.txt`	01/22/2026 9:54 AM UTC