docker: only import nvidia-persitenced.service if nvidia-ctk is insta…

…lled
saltyorg · Feb 18, 2024 · c663f66 · c663f66
1 parent e4185b9
commit c663f66
Showing 1 changed file with 25 additions and 17 deletions.
diff --git a/roles/docker/tasks/main.yml b/roles/docker/tasks/main.yml
@@ -78,23 +78,31 @@
 - name: Nvidia task block
   when: gpu.nvidia
   block:
-    - name: Import 'nvidia-persistenced.service'
-      ansible.builtin.copy:
-        src: nvidia-persistenced.service
-        dest: "/etc/systemd/system/nvidia-persistenced.service"
-        owner: "root"
-        group: "root"
-        mode: "0644"
-
-    - name: Run systemd daemon-reload
-      ansible.builtin.systemd_service:
-        daemon_reload: true
-
-    - name: Start nvidia-persistenced service
-      ansible.builtin.systemd_service:
-        name: nvidia-persistenced.service
-        state: started
-        enabled: true
+    - name: Check if '/usr/bin/nvidia-ctk' exists
+      ansible.builtin.stat:
+        path: "/usr/bin/nvidia-ctk"
+      register: nvidia_ctk
+
+    - name: Tasks for when nvidia drivers are installed
+      when: nvidia_ctk.stat.exists
+      block:
+        - name: Import 'nvidia-persistenced.service'
+          ansible.builtin.copy:
+            src: nvidia-persistenced.service
+            dest: "/etc/systemd/system/nvidia-persistenced.service"
+            owner: "root"
+            group: "root"
+            mode: "0644"
+
+        - name: Run systemd daemon-reload
+          ansible.builtin.systemd_service:
+            daemon_reload: true
+
+        - name: Start nvidia-persistenced service
+          ansible.builtin.systemd_service:
+            name: nvidia-persistenced.service
+            state: started
+            enabled: true
 
 - name: "Import Docker Binary tasks"
   ansible.builtin.include_tasks: "subtasks/binary/binary.yml"