- name: Add NVIDIA Container Toolkit repository
  copy:
    src: nvidia-container-toolkit.repo
    dest: /etc/yum.repos.d/nvidia-container-toolkit.repo
    mode: '0644'
  become: true

- name: Enable NVIDIA Container Toolkit experimental repository
  shell: yum-config-manager --enable nvidia-container-toolkit-experimental
  become: true
  args:
    executable: /bin/bash

- name: Install NVIDIA Container Toolkit
  package:
    name: nvidia-container-toolkit-1.17.8-1
    state: present
  become: true