---
- name: Install NVIDIA driver package
  ansible.builtin.apt:
    name: "{{ nvidia_runtime_selected_driver }}"
    state: present
    update_cache: true
  when: nvidia_runtime_install_driver | bool

- name: Install CUDA toolkit package
  ansible.builtin.apt:
    name: "{{ nvidia_runtime_cuda_package }}"
    state: present
  when: nvidia_runtime_install_cuda_toolkit | bool

- name: Install NVIDIA container toolkit package
  ansible.builtin.apt:
    name: "{{ nvidia_runtime_container_toolkit_package }}"
    state: present
  when: nvidia_runtime_install_container_toolkit | bool