Update NVIDIA container toolkit repository and restart docker daemon

autowarefoundation · Jan 31, 2024 · 66fe550 · 66fe550
1 parent cc7342a
commit 66fe550
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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