Fix wrong nvidia gpu card number (#5353)

microsoft · Mar 8, 2021 · 648d1bc · 648d1bc
1 parent f0dad8b
commit 648d1bc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/contrib/kubespray/roles/requirement/computing-devices/nvidia.com_gpu/tasks/main.yml b/contrib/kubespray/roles/requirement/computing-devices/nvidia.com_gpu/tasks/main.yml
@@ -46,14 +46,14 @@
 
 - name: "Check NVIDIA GPU card number is matched or not"
   set_fact:
-    unmet_requirements: "{{ unmet_requirements + [\"NVIDIA GPU card number is not matched: {{ computing_device_count }} specified but only {{ nvidia_gpu_count.stdout_lines[0] }} found\"] }}"
+    unmet_requirements: "{{ unmet_requirements + [\"NVIDIA GPU card number is not matched: {{ computing_device_count }} specified but {{ nvidia_gpu_count.stdout|trim }} found\"] }}"
   changed_when: false
   check_mode: false
   environment: {}
   when:
     # if nvidia-smi doesn't work, we can skip this step
     - nvidia_smi.rc == 0
-    - "nvidia_gpu_count.stdout_lines[0]|int != computing_device_count"
+    - 'nvidia_gpu_count.stdout|trim|int != computing_device_count'
     - nvidia_gpu_count.rc == 0
 
 - name: "Check default docker runtime"