---
# ------------------------------------------------------------------------------
# FILE: roles/llm-inference/tasks/driver.yml
# DESCRIPTION: Phase 2 — NVIDIA driver.
#              Installs nvidia-driver-595-open via apt. Fully idempotent —
#              already installed on astro-orbiter on 2026-08-03, this is a no-op.
#              DKMS builds the kernel module automatically on install.
# ------------------------------------------------------------------------------

- name: Install NVIDIA driver package
  ansible.builtin.apt:
    name: "{{ llm_nvidia_driver_package }}"
    state: present
    update_cache: false
  notify: reload systemd

- name: Verify nvidia-smi reports the GPU
  ansible.builtin.command: nvidia-smi --query-gpu=name,memory.total,driver_version --format=csv,noheader
  register: nvidia_smi_out
  changed_when: false
  failed_when: nvidia_smi_out.rc != 0

- name: Print nvidia-smi output
  ansible.builtin.debug:
    msg: "GPU detected: {{ nvidia_smi_out.stdout }}"