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
fe1cc74e9eaa87328b111679d4ef4c0c1167a8659574d2771ad1c2eb56cce0feCONFIG
Entrypoint /opt/nvidia/nvidia_entrypoint.sh; WorkingDir /workspace; ExposedPorts 6006/tcp, 8888/tcp
03/11/2025 6:48 PM UTC
5f19f5e33c15f963b2769a2786c73c2511bc66bd77af8c7271e7340c1589f2fdARG
NVIDIA_BUILD_REF=84f446f5c7204f47f7769a5a5f759c7622b828eb
03/11/2025 6:48 PM UTC
3d801c33cd498eae369cf8f16c2c1bc59509ac3a4e714a6789e706e5d9165d51LABEL
com.nvidia.build.id=148941828
03/11/2025 6:48 PM UTC
7ad4d27ba1de2a456879efd185df78a0512bc2e691ea3710912487c0658510d4ENV
NVIDIA_BUILD_ID=148941828
03/11/2025 6:48 PM UTC
e92a417ca959c89611ad1ed8089fd271af0c87b11c95631afc526fdae3cf50f2ARG
NVIDIA_BUILD_ID=148941828
03/11/2025 6:48 PM UTC
9ef6a21bb768c9f537e4b4f53ec2b02a54c5ce0c418bad01648f7aab41378c1eRUN
NVIDIA_PYTORCH_VERSION=25.03 PYTORCH_BUILD_VERSION=2.7.0a0+7c8ec84 NVFUSER_BUILD_VERSION=c5e1555 TARGETARCH=amd64 PYVER=3.12 PYVER_MAJMIN=312 L4T=0 DALI_EXTRA_INDEX_URL=http://sqrl/nvdl/datasets/dali/pip-dali cd /opt/pytorch/pytorch/third_party/gloo &&
  patch -p1 < /gloo.patch &&
  cd - &&
  rm /gloo.patch
03/11/2025 6:48 PM UTC
ca9bfa38910e87a011067434800bd96b96b9b2044d430356aeb8022057ea539cCOPY
gloo.patch /gloo.patch
03/11/2025 6:48 PM UTC
264a93db8052f49ab226963dc28da4bacd35c543a9b0014efc414a6b0e7239a2COPY
entrypoint.d/ /opt/nvidia/entrypoint.d/
03/11/2025 6:48 PM UTC
97c2dad5f9a2c40bbf4c84970eb5a5194069603b6118386f9504d7b92d6e23acRUN
NVIDIA_PYTORCH_VERSION=25.03 PYTORCH_BUILD_VERSION=2.7.0a0+7c8ec84 NVFUSER_BUILD_VERSION=c5e1555 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
03/11/2025 6:48 PM UTC
29f831f42fbe6d2138f001a2b7937d0f12c1faae78cd35ebe91f7a487efe02e5ENV
TORCH_NCCL_USE_COMM_NONBLOCKING=0
03/11/2025 6:48 PM UTC
...