ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
NVIDIA
NVIDIA
PyTorch
Container
NVIDIA
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
LayerLabelCreated
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.ref=d557151f4c7ddca284cb5e8d5ce78cee4d80f7e5
06/09/2026 11:57 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_REF=d557151f4c7ddca284cb5e8d5ce78cee4d80f7e5
06/09/2026 11:57 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.id=337426143
06/09/2026 11:57 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NVIDIA_BUILD_ID=337426143
06/09/2026 11:57 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_ID=337426143
06/09/2026 11:57 PM UTC
sha256:4f4fb700ef54461cfa02571ae0db9a0dc1e0cdb5577484a6d75e68dc38e8acc1RUN
NVIDIA_PYTORCH_VERSION=26.06 PYTORCH_BUILD_VERSION=2.13.0a0+8145d63 NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+git28e6aca5 TORCHTITAN_BUILD_VERSION=0.2.2+git2ae13405 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
06/09/2026 11:57 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
ENABLE_FIPS=0
06/09/2026 11:57 PM UTC
sha256:205aaae67b171d3d7b40415cc0d451f0dcb8b44f3ed64002cda2be4b731558d3COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
06/09/2026 11:57 PM UTC
sha256:a6d4143271c8dfa9455d802f68c88446595721b2bc7d00222cb8e0c3c31f5aabRUN
NVIDIA_PYTORCH_VERSION=26.06 PYTORCH_BUILD_VERSION=2.13.0a0+8145d63 NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+git28e6aca5 TORCHTITAN_BUILD_VERSION=0.2.2+git2ae13405 TARGETARCH=amd64 PYVER=3.12 ENABLE_MITMPROXY=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali cp /tmp/restricted_constraint.txt /etc/pip/constraint.txt &&
  cp /tmp/original_constraint.txt /etc/pip/original_constraint.txt
06/09/2026 11:57 PM UTC
sha256:5955932b0687879336a526f18bf266ec32d61fe7c1ff41c39772a917abc8297aRUN
NVIDIA_PYTORCH_VERSION=26.06 PYTORCH_BUILD_VERSION=2.13.0a0+8145d63 NVFUSER_BUILD_VERSION=0.1.4a0+nvidia TORCHAO_BUILD_VERSION=+git28e6aca5 TORCHTITAN_BUILD_VERSION=0.2.2+git2ae13405 TARGETARCH=amd64 PYVER=3.12 ENABLE_MITMPROXY=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali 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
06/09/2026 11:57 PM UTC
...