File tree Expand file tree Collapse file tree
examples/machine-learning Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -184,11 +184,12 @@ deployment_groups:
184184 apt update -y
185185 apt install -y cuda-toolkit-12-8
186186 apt install -y nvidia-container-toolkit
187- apt install -y datacenter-gpu-manager=1:4.5.0
188- apt install -y datacenter-gpu-manager-dev=1:4.5.0
187+ #Pin the package to latest DCGMI version 4.5.0
188+ apt install -y datacenter-gpu-manager-4-cuda12=1:4.5.0-1
189+ apt install -y datacenter-gpu-manager-4-dev=1:4.5.0-1
189190
190191 #Hold packages to prevent auto-updates
191- apt-mark hold datacenter-gpu-manager datacenter-gpu-manager-dev
192+ apt-mark hold datacenter-gpu-manager-4-cuda12 datacenter-gpu-manager-4 -dev
192193 # this duplicates the ulimits configuration of the HPC VM Image
193194 - type : data
194195 destination : /etc/security/limits.d/99-unlimited.conf
Original file line number Diff line number Diff line change @@ -479,8 +479,9 @@ deployment_groups:
479479 enable_nvidia_dcgm: $(vars.enable_nvidia_dcgm)
480480 nvidia_packages:
481481 - cuda-toolkit-12-8
482- - datacenter-gpu-manager=1:4.5.0
483- - datacenter-gpu-manager-dev=1:4.5.0
482+ #Pin the package to latest DCGMI version 4.5.0
483+ - datacenter-gpu-manager-4-cuda12=1:4.5.0-1
484+ - datacenter-gpu-manager-4-dev=1:4.5.0-1
484485 - libnvidia-cfg1-570-server
485486 - libnvidia-nscq-570
486487 - nvidia-compute-utils-570-server
Original file line number Diff line number Diff line change @@ -159,11 +159,11 @@ deployment_groups:
159159 set -e -o pipefail
160160 add-nvidia-repositories -y
161161 apt install -y cuda-toolkit-12-8
162- apt install -y datacenter-gpu-manager=1: 4.5.0
163- apt install -y datacenter-gpu-manager-dev =1:4.5.0
164-
162+ #Pin the package to latest DCGMI version 4.5.0
163+ apt install -y datacenter-gpu-manager-4-cuda12 =1:4.5.0-1
164+ apt install -y datacenter-gpu-manager-4-dev=1:4.5.0-1
165165 #Hold packages to prevent auto-updates
166- apt-mark hold datacenter-gpu-manager datacenter-gpu-manager-dev
166+ apt-mark hold datacenter-gpu-manager-4-cuda12 datacenter-gpu-manager-4 -dev
167167 - type : ansible-local
168168 destination : settings_nvidia_dcgm.yml
169169 content : |
Original file line number Diff line number Diff line change @@ -161,11 +161,12 @@ deployment_groups:
161161 add-nvidia-repositories -y
162162 apt update -y
163163 apt install -y cuda-toolkit-12-8
164- apt install -y datacenter-gpu-manager=1:4.5.0
165- apt install -y datacenter-gpu-manager-dev=1:4.5.0
166-
164+ #Pin the package to latest DCGMI version 4.5.0
165+ apt install -y datacenter-gpu-manager-4-cuda12=1:4.5.0-1
166+ apt install -y datacenter-gpu-manager-4-dev=1:4.5.0-1
167+
167168 #Hold packages to prevent auto-updates
168- apt-mark hold datacenter-gpu-manager datacenter-gpu-manager-dev
169+ apt-mark hold datacenter-gpu-manager-4-cuda12 datacenter-gpu-manager-4 -dev
169170 - type : ansible-local
170171 destination : configure_cuda_dcgm.yml
171172 content : |
Original file line number Diff line number Diff line change @@ -177,8 +177,9 @@ deployment_groups:
177177 cuda_repo_filename: /tmp/{{ cuda_repo_url | basename }}
178178 nvidia_packages:
179179 - cuda-toolkit-12-8
180- - datacenter-gpu-manager=1:4.5.0
181- - datacenter-gpu-manager-dev=1:4.5.0
180+ #Pin the package to latest DCGMI version 4.5.0
181+ - datacenter-gpu-manager-4-cuda12=1:4.5.0-1
182+ - datacenter-gpu-manager-4-dev=1:4.5.0-1
182183 tasks:
183184 - name: Download NVIDIA repository package
184185 ansible.builtin.get_url:
You can’t perform that action at this time.
0 commit comments