software-infrastructure/roles/ericomeehan.nvidia_driver_debian/tasks/main.yml
2024-07-04 19:04:55 +00:00

33 lines
737 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 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