diff --git a/ansible/roles/nvidia_docker/tasks/main.yaml b/ansible/roles/nvidia_docker/tasks/main.yaml index 6a0fffa2468..34ae045cc2c 100644 --- a/ansible/roles/nvidia_docker/tasks/main.yaml +++ b/ansible/roles/nvidia_docker/tasks/main.yaml @@ -8,7 +8,7 @@ - name: Add NVIDIA container toolkit repository become: true ansible.builtin.apt_repository: - repo: deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list + repo: deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://nvidia.github.io/libnvidia-container/stable/deb/$(ARCH) / state: present filename: nvidia-container-toolkit update_cache: true @@ -25,3 +25,9 @@ ansible.builtin.shell: | nvidia-ctk runtime configure --runtime=docker changed_when: true + +- name: Restart docker daemon + become: true + ansible.builtin.shell: | + systemctl restart docker + changed_when: true