v0.0.6

roles/ericomeehan.nvidia_driver_debian/tasks/main.yml

@@ -5,28 +5,31 @@
     dest: /etc/apt/sources.list
     regexp: '^(deb(?!.* contrib).*)'
     replace: '\1 contrib non-free'
+
 - name: Update apt
   become: yes
   apt:
     update_cache: yes
   when: nvidia == true
+
 - name: Install Linux headers
   apt:
     name: linux-headers-{{ ansible_kernel }}
     state: present
-- name: Install Nvidia driver
+
+- name: Install Nvidia and CUDA drivers
   apt:
-    name: nvidia-driver
-    state: present
-- name: Install necessary firmware
-  apt:
-    name: firmware-misc-nonfree
-    state: present
-- name: Install CUDA drivers
-  apt:
-    name: nvidia-cuda-dev 
-    state: present
-- name: Install CUDA toolkit
-  apt:
-    name: nvidia-cuda-toolkit
     state: present
+    name: 
+      - nvidia-driver
+      - firmware-misc-nonfree
+      - nvidia-cuda-dev
+      - nvidia-cuda-toolkit
+
+- name: Rebooting the system
+  ansible.builtin.reboot:
+
+- name: Wait for SSH connection
+  ansible.builtin.wait_for_connection:
+    timeout: 300
+