21 lines
751 B
YAML
21 lines
751 B
YAML
---
|
|
- name: Check nvidia-smi status
|
|
ansible.builtin.command: nvidia-smi --query-gpu=name,memory.total,driver_version --format=csv,noheader
|
|
register: nvidia_runtime_smi
|
|
changed_when: false
|
|
failed_when: false
|
|
|
|
- name: Fail when post-install validation is required and nvidia-smi is unavailable
|
|
ansible.builtin.fail:
|
|
msg: >-
|
|
nvidia-smi is unavailable after installation.
|
|
This usually means a reboot is still required or the selected driver is incompatible.
|
|
when:
|
|
- nvidia_runtime_validate_after_install | bool
|
|
- nvidia_runtime_smi.rc != 0
|
|
|
|
- name: Capture GPU info lines
|
|
ansible.builtin.set_fact:
|
|
nvidia_runtime_gpu_lines: "{{ nvidia_runtime_smi.stdout_lines | default([]) }}"
|
|
when: nvidia_runtime_smi.rc == 0
|