Skip to content

Commit 67c383d

Browse files
Pin DCGM version for A3H
1 parent 4b471a2 commit 67c383d

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

examples/machine-learning/a3-highgpu-8g/a3high-slurm-blueprint.yaml

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,9 @@ deployment_groups:
193193
- cuda-toolkit-12-8
194194
- nvidia-dkms-580-server-open
195195
- nvidia-driver-580-server-open
196-
- datacenter-gpu-manager-4-cuda13
196+
- datacenter-gpu-manager-4-cuda13=1:4.6.0-1
197+
- datacenter-gpu-manager-4-dev=1:4.6.0-1
198+
- datacenter-gpu-manager-4-core=1:4.6.0-1
197199
- libnvidia-nscq-580
198200
- nvidia-container-toolkit
199201
tasks:
@@ -209,10 +211,11 @@ deployment_groups:
209211
ansible.builtin.apt:
210212
name: "{{ item }}"
211213
update_cache: true
214+
allow_downgrade: yes
212215
loop: "{{ nvidia_packages }}"
213216
- name: Freeze NVIDIA drivers and CUDA
214217
ansible.builtin.dpkg_selections:
215-
name: "{{ item }}"
218+
name: "{{ item.split('=')[0] }}"
216219
selection: hold
217220
loop: "{{ nvidia_packages }}"
218221
- name: Create nvidia-persistenced override directory

0 commit comments

Comments
 (0)