v0.0.8

roles/ericomeehan.nvidia_driver_debian/defaults/main.yml

@@ -1,3 +1,5 @@
 ---
 # defaults file for ericomeehan.nvidia_driver_debian
 nvidia_driver_skip_reboot: false
+nvidia_driver_source: nvidia
+nvidia_driver_tesla: false

roles/ericomeehan.nvidia_driver_debian/tasks/main.yml

@@ -1,31 +1,9 @@
 ---
 # 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: Install Nvidia proprietary drivers
+  include_tasks: nvidia-proprietary.yml
+  when: nvidia_driver_source == "nvidia"
 
-- name: Update apt
-  become: yes
-  apt:
-    update_cache: yes
-
-- name: Install Linux headers
-  apt:
-    name: linux-headers-{{ ansible_kernel }}
-    state: present
-
-- name: Install Nvidia and CUDA drivers
-  apt:
-    state: present
-    name: 
-      - nvidia-driver
-      - firmware-misc-nonfree
-      - nvidia-cuda-dev
-      - nvidia-cuda-toolkit
-    register: install_driver
-
-- name: Reboot after driver install
-  reboot:
-  when: install_driver.changed and not nvidia_driver_skip_reboot
+- name: Install nouveau and mesa drivers
+  include_tasks: nouveau-mesa.yml
+  when: nvidia_driver_source == "nouveau"

roles/ericomeehan.nvidia_driver_debian/tasks/nouveau-mesa.yml

@@ -0,0 +1,18 @@
+---
+# tasks file for nouveau-mesa
+- name: Update apt
+  become: yes
+  apt:
+    update_cache: yes
+
+- name: Install nouveau and mesa
+  apt:
+    state: present
+    name:
+      - xserver-xorg-video-nouveau
+      - mesa-utils
+    register: install_driver
+
+- name: Reboot after driver install
+  reboot:
+  when: install_driver.changed and not nvidia_driver_skip_reboot

roles/ericomeehan.nvidia_driver_debian/tasks/nvidia-proprietary.yml

@@ -0,0 +1,38 @@
+---
+# tasks file for nvidia-proprietary
+- 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
+
+- name: Install Linux headers
+  apt:
+    name: linux-headers-{{ ansible_kernel }}
+    state: present
+
+- name: Install Nvidia and CUDA drivers
+  apt:
+    state: present
+    name: 
+      - nvidia-driver
+      - firmware-misc-nonfree
+      - nvidia-cuda-dev
+      - nvidia-cuda-toolkit
+  register: install_driver
+
+- name: Install tesla drivers
+  apt:
+    state: present
+    name:
+      - nvidia-tesla-470-driver
+  when: nvidia_driver_tesla == true
+
+- name: Reboot after driver install
+  reboot:
+  when: install_driver.changed and not nvidia_driver_skip_reboot