Freezing the DCGMI Version To Latest Version 4.5.0

Author: LAVEEN

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

@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2026 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -184,8 +184,12 @@ deployment_groups:
           apt update -y
           apt install -y cuda-toolkit-12-8
           apt install -y nvidia-container-toolkit
-          apt install -y datacenter-gpu-manager-4-cuda12
-          apt install -y datacenter-gpu-manager-4-dev
+          apt install -y datacenter-gpu-manager=1:4.5.0
+          apt install -y datacenter-gpu-manager-dev=1:4.5.0
+
+          # We Adding apt-mark hold to "freeze" the version so it 
+          # doesn't auto-update during future 'apt upgrade' calls.
+          apt-mark hold datacenter-gpu-manager datacenter-gpu-manager-dev
       # this duplicates the ulimits configuration of the HPC VM Image
       - type: data
         destination: /etc/security/limits.d/99-unlimited.conf

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

@@ -1,4 +1,4 @@
-# Copyright 2025 "Google LLC"
+# Copyright 2026 "Google LLC"
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

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

@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2026 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -479,8 +479,8 @@ deployment_groups:
               enable_nvidia_dcgm: $(vars.enable_nvidia_dcgm)
               nvidia_packages:
               - cuda-toolkit-12-8
-              - datacenter-gpu-manager-4-cuda12
-              - datacenter-gpu-manager-4-dev
+              - datacenter-gpu-manager=1:4.5.0
+              - datacenter-gpu-manager-dev=1:4.5.0
               - libnvidia-cfg1-570-server
               - libnvidia-nscq-570
               - nvidia-compute-utils-570-server

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

@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2026 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -159,8 +159,11 @@ deployment_groups:
           set -e -o pipefail
           add-nvidia-repositories -y
           apt install -y cuda-toolkit-12-8
-          apt install -y datacenter-gpu-manager-4-cuda12
-          apt install -y datacenter-gpu-manager-4-dev
+          apt install -y datacenter-gpu-manager=1:4.5.0
+          apt install -y datacenter-gpu-manager-dev=1:4.5.0
+          
+          #Hold packages to prevent auto-updates
+          apt-mark hold datacenter-gpu-manager datacenter-gpu-manager-dev
       - type: ansible-local
         destination: settings_nvidia_dcgm.yml
         content: |

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

@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2026 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

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

@@ -1,4 +1,4 @@
-# Copyright 2024 Google LLC
+# Copyright 2026 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

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

@@ -1,4 +1,4 @@
-# Copyright 2025 Google LLC
+# Copyright 2026 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -161,8 +161,10 @@ deployment_groups:
           add-nvidia-repositories -y
           apt update -y
           apt install -y cuda-toolkit-12-8
-          apt install -y datacenter-gpu-manager-4-cuda12
-          apt install -y datacenter-gpu-manager-4-dev
+          apt install -y datacenter-gpu-manager=1:4.5.0
+          apt install -y datacenter-gpu-manager-dev=1:4.5.0
+          # Pin the packages so they aren't upgraded by 'apt upgrade'
+          apt-mark hold datacenter-gpu-manager datacenter-gpu-manager-dev
       - type: ansible-local
         destination: configure_cuda_dcgm.yml
         content: |

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

@@ -1,4 +1,4 @@
-# Copyright 2025 Google LLC
+# Copyright 2026 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

examples/machine-learning/a4-highgpu-8g/a4high-vm-deployment.yaml

@@ -1,4 +1,4 @@
-# Copyright 2025 Google LLC
+# Copyright 2026 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

examples/machine-learning/a4x-highgpu-4g/a4xhigh-slurm-blueprint.yaml

@@ -1,4 +1,4 @@
-# Copyright 2025 Google LLC
+# Copyright 2026 Google LLC
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -177,8 +177,8 @@ deployment_groups:
               cuda_repo_filename: /tmp/{{ cuda_repo_url | basename }}
               nvidia_packages:
               - cuda-toolkit-12-8
-              - datacenter-gpu-manager-4-cuda12
-              - datacenter-gpu-manager-4-dev
+              - datacenter-gpu-manager=1:4.5.0
+              - datacenter-gpu-manager-dev=1:4.5.0
             tasks:
             - name: Download NVIDIA repository package
               ansible.builtin.get_url: