v0.0.7

roles/ericomeehan.nvidia_driver_debian/defaults/main.yml

@@ -1,2 +1,3 @@
 ---
 # defaults file for ericomeehan.nvidia_driver_debian
+nvidia_driver_skip_reboot: false

roles/ericomeehan.nvidia_driver_debian/tasks/main.yml

@@ -10,7 +10,6 @@
   become: yes
   apt:
     update_cache: yes
-  when: nvidia == true
 
 - name: Install Linux headers
   apt:
@@ -25,11 +24,8 @@
       - firmware-misc-nonfree
       - nvidia-cuda-dev
       - nvidia-cuda-toolkit
+    register: install_driver
 
-- name: Rebooting the system
-  ansible.builtin.reboot:
-
-- name: Wait for SSH connection
-  ansible.builtin.wait_for_connection:
-    timeout: 300
-
+- name: Reboot after driver install
+  reboot:
+  when: install_driver.changed and not nvidia_driver_skip_reboot