17 lines
559 B
YAML
17 lines
559 B
YAML
|
---
|
||
|
- name: Setup NVIDIA drivers
|
||
|
become: true
|
||
|
tasks:
|
||
|
- name: "Adding NVIDIA CUDA repository"
|
||
|
ansible.builtin.apt_repository:
|
||
|
repo: "deb [signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
|
||
|
filename: nvidia-upstream-ppa
|
||
|
- name: "Installing CUDA"
|
||
|
ansible.builtin.apt:
|
||
|
pkg:
|
||
|
- cuda-12
|
||
|
- nvidia-headless
|
||
|
- name: "Making sure NVIDIA UVM is working"
|
||
|
community.general.modprobe:
|
||
|
name: "nvidia_uvm"
|
||
|
state: present
|