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
0c284ccd34247eec5a3f77de4878c3b77bf252021ab81947bf5525f2d272e1e5	CONFIG	`Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp`	05/07/2026 11:34 PM UTC
46cde424b2f625b9e8543dbce9e7cd460703af14fee6d082b781fff338a37a22	ARG	`NVIDIA_BUILD_REF=30a5fc6cbfce157e75fae3d0cf1fd8e273a3dc25`	05/07/2026 11:34 PM UTC
15602d8593fefce39d0e86bd872f320805abd8e95d373ea9998349d8dc1db139	LABEL	`com.nvidia.build.id=313520559`	05/07/2026 11:34 PM UTC
63208f082bc10f9543748b049dd6735ab08a5d346d85e7ffedcc8515b4cc8af7	ENV	`NVIDIA_BUILD_ID=313520559`	05/07/2026 11:34 PM UTC
5b1d590c9d4256d92e9959c9e4b56d678942d3684bae25d6a99625e79aa8a6f3	ARG	`NVIDIA_BUILD_ID=313520559`	05/07/2026 11:34 PM UTC
3ef0a26cd34ae02a781068f36906932cc827e4391c33602fc69501fa2c8c4c5e	RUN	`NVIDIA_PYTORCH_VERSION=26.05 PYTORCH_BUILD_VERSION=2.12.0a0+5aff392 NVFUSER_BUILD_VERSION=29f5740 TORCHAO_BUILD_VERSION=+git8781a869 TORCHTITAN_BUILD_VERSION=0.2.2+gita00ef7df TARGETARCH=amd64 PYVER=3.12 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`	05/07/2026 11:34 PM UTC
41b018b029c4542ffee4571f2addb75107efda4ef42268a2eff4f670ae11ee5e	ARG	`ENABLE_FIPS=0`	05/07/2026 11:34 PM UTC
afd805fd119fdb9d230abc7a84541255084fe4b83102c3f5d948f29eef09546a	COPY	`entrypoint.d/ /opt/nvidia/entrypoint.d/`	05/07/2026 11:34 PM UTC
8711fb5a2a971e72788c2d5da71b4b02b5ac57d92aa07e5f0e25e7fa897fa080	COPY	`constraint.txt /etc/pip/original_constraint.txt`	05/07/2026 11:34 PM UTC
f98288265dc768fb980a05766637221c583cc7d3627f9ea5120248d69a4ea518	COPY	`restricted_constraint.txt /etc/pip/constraint.txt`	05/07/2026 11:34 PM UTC