---
- name: Add nvidia-docker GPG apt-key
  apt_key:
    url: https://nvidia.github.io/nvidia-docker/gpgkey
    id: C95B321B61E88C1809C4F759DDCAE044F796ECB0
  become: yes

- name: Add nvidia-docker repository in /etc/apt/sources.list.d
  get_url:
    url: "https://nvidia.github.io/nvidia-docker/{{ ansible_distribution | lower }}{{ ansible_distribution_version }}//nvidia-docker.list"
    dest: /etc/apt/sources.list.d/nvidia-docker.list
    checksum: sha256:1727985494fbd19e3b963880d15117487435cbabef4e295484111f003cf03d41
  become: yes

- name: Update and install nvidia-container-toolkit
  apt:
    name: nvidia-container-toolkit
    update_cache: yes
  become: yes