---
# 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 and CUDA drivers
  apt:
    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