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
375fd777fc6f9b688f55af9563554be247203811716303596d86695caff9d062CONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
06/12/2025 4:08 PM UTC
6f97ac2f39ed767f0ec6e172307a8e015fc41f3194917f9296cdcfd340fdbe02ARG
NVIDIA_BUILD_REF=7b41d3f55736e5e072de5b4de2ea6dace7df1ca8
06/12/2025 4:08 PM UTC
b5d654cd66f21aa1dbc71845232b4f5719c74b2027843692ba137b5d005216e8LABEL
com.nvidia.build.id=177567386
06/12/2025 4:08 PM UTC
0ef8987d13bf15de3326bda83321ef6474f5cf823c312db5c7ad4f1e3c71fa41ENV
NVIDIA_BUILD_ID=177567386
06/12/2025 4:08 PM UTC
da61f3469b5d5627e090df0e7516de1e83f94c45227e22b5994afc7a1f282806ARG
NVIDIA_BUILD_ID=177567386
06/12/2025 4:08 PM UTC
b102ebbd5cf296d1b17366691cfdbe93283ee7a3672e7ede30102c3389bb55fbCOPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
06/12/2025 4:08 PM UTC
4d8e808776d87d00414d8081f1e426baf139d122cb4035197d96bbd2b85c0803COPY
constraint.txt /etc/pip/original_constraint.txt
06/12/2025 4:08 PM UTC
33112ad14117810c31559039ef0629fdc7c629be47ba339e913ecd821845579fCOPY
restricted_constraint.txt /etc/pip/constraint.txt
06/12/2025 4:08 PM UTC
093d82e3943de42c00590599db400856141e444545629b59e70f1996d88b45faRUN
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
ff30388c1927c19e9932f50b1e421144e56464e903743ddb8b747a109b9b3131ENV
TORCH_NCCL_USE_COMM_NONBLOCKING=0
06/12/2025 4:08 PM UTC
...