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