- name: install nvidia-container-toolkit repo
  become: true
  ansible.builtin.shell:
    cmd: "curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --yes --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | tee /etc/apt/sources.list.d/nvidia-container-toolkit.list"

- name: install Nvidia driver
  become: true
  ansible.builtin.apt:
    pkg:
      - nvidia-container-toolkit
      - nvidia-driver
      - firmware-misc-nonfree
    state: present

- name: configure nvidia for rootless podman
  become: true
  ansible.builtin.shell:
    cmd: nvidia-ctk cdi generate --output=/etc/cdi/nvidia.yaml
  register: nvidia_ctk_result
  ignore_errors: true
  failed_when: nvidia_ctk+result.rc != 1