Merge pull request GoogleCloudPlatform#4228 from CIeNET-International/dcgm

Update to datacenter-gpu-manager-4 package in A-series blueprints

Author: tpdownes

examples/machine-learning/a3-highgpu-8g/ml-slurm-a3-1-image.yaml

@@ -173,7 +173,7 @@ deployment_groups:
             - name: Install NVIDIA DCGM
               ansible.builtin.apt:
                 name:
-                - datacenter-gpu-manager
+                - datacenter-gpu-manager-4-cuda12
                 - libnvidia-nscq-550
                 update_cache: true
             post_tasks:

examples/machine-learning/a3-megagpu-8g/a3mega-slurm-blueprint.yaml

@@ -207,7 +207,7 @@ deployment_groups:
               enable_nvidia_dcgm: $(vars.enable_nvidia_dcgm)
               nvidia_packages:
               - cuda-toolkit-12-8
-              - datacenter-gpu-manager
+              - datacenter-gpu-manager-4-cuda12
               - libnvidia-cfg1-570-server
               - libnvidia-nscq-570
               - nvidia-compute-utils-570-server

examples/machine-learning/a3-ultragpu-8g/a3ultra-slurm-blueprint.yaml

@@ -134,7 +134,7 @@ deployment_groups:
               enable_nvidia_dcgm: false
               nvidia_packages:
               - cuda-toolkit-12-8
-              - datacenter-gpu-manager
+              - datacenter-gpu-manager-4-cuda12
               - libnvidia-cfg1-570-server
               - libnvidia-nscq-570
               - nvidia-compute-utils-570-server

examples/machine-learning/a4-highgpu-8g/a4high-slurm-blueprint.yaml

@@ -135,7 +135,7 @@ deployment_groups:
               enable_nvidia_dcgm: false
               nvidia_packages:
               - cuda-toolkit-12-8
-              - datacenter-gpu-manager
+              - datacenter-gpu-manager-4-cuda12
               - libnvidia-cfg1-570-server
               - libnvidia-nscq-570
               - nvidia-compute-utils-570-server