---
- name: Import NVIDIA Docker repository gpg keys
  rpm_key:
    key: "{{ nvidia_docker_repo_url }}/{{ item }}/gpgkey"
    state: present
  with_items: "{{ nvidia_repos }}"
  when: "'cuda' in group_names"

- name: Configure Nvidia repositories (yum)
  yum_repository:
    name: "{{ item }}"
    description: Official {{ item }} repository
    baseurl: "{{ nvidia_docker_repo_url }}/{{ item }}/{{ nvidia_docker_repo_subfolder }}/x86_64/"
    gpgkey: "{{ nvidia_docker_repo_url }}/{{ item }}/gpgkey"
    gpgcheck: no
    enabled: yes
  with_items: "{{ nvidia_repos }}"
  when: "'cuda' in group_names"