36 lines
727 B
YAML
36 lines
727 B
YAML
---
|
|
# 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
|
|
|