---
# 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