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=7b41d3f55736e5e072de5b4de2ea6dace7df1ca8
06/12/2025 4:08 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_REF=7b41d3f55736e5e072de5b4de2ea6dace7df1ca8
06/12/2025 4:08 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4LABEL
com.nvidia.build.id=177567386
06/12/2025 4:08 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
NVIDIA_BUILD_ID=177567386
06/12/2025 4:08 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
NVIDIA_BUILD_ID=177567386
06/12/2025 4:08 PM UTC
sha256:b9ad851e1bef0c80fd03c06ebd62daba11ac4d79cb5d2d26acea2c0320474852COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
06/12/2025 4:08 PM UTC
sha256:27872fc3573e1c2d159cd41a4db30d77bfe8ac1963b87567dc78ad58674d71a5COPY
constraint.txt /etc/pip/original_constraint.txt
06/12/2025 4:08 PM UTC
sha256:e44523bf20109372c6636c82b13233e767d5e725e086704f4d08112690d4c6a6COPY
restricted_constraint.txt /etc/pip/constraint.txt
06/12/2025 4:08 PM UTC
sha256:a25409264d46a2ce98edb8ec59308b7bc56fd8af86825741f6d7e2f401145e32RUN
NVIDIA_PYTORCH_VERSION=25.06 PYTORCH_BUILD_VERSION=2.8.0a0+5228986 NVFUSER_BUILD_VERSION=9bf5aca TARGETARCH=amd64 PYVER=3.12 PYVER_MAJMIN=312 L4T=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/12/2025 4:08 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
TORCH_NCCL_USE_COMM_NONBLOCKING=0
06/12/2025 4:08 PM UTC
...