- name: Gather current kernel version
  ansible.builtin.shell: uname -r
  register: kernel_version

- name: install Nvidia driver
  become: true
  ansible.builtin.apt:
    pkg:
      - nvidia-driver
      - firmware-misc-nonfree
      - nvidia-driver-libs:i386
    state: present
    update_cache: true
  register: updated_driver

- name: Unconditionally reboot the machine with all defaults
  become: true
  ansible.builtin.reboot:
  when: updated_driver.changed