39 lines
818 B
YAML
39 lines
818 B
YAML
---
|
|
# 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
|