v0.0.5

roles/ericomeehan.nvidia_driver_debian/tasks/main.yml

@@ -0,0 +1,32 @@
+---
+# tasks file for ericomeehan.nvidia_driver_debian
+- name: Add contrib & non-free repository
+  replace:
+    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
+  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