Bart Sjerps
51 lines
1.3 KiB
YAML
51 lines
1.3 KiB
YAML
---
|
|
- name: Setup Nvidia GPU drivers
|
|
hosts: nvidia
|
|
tasks:
|
|
- name: nVidia CUDA Repo
|
|
ansible.builtin.yum_repository:
|
|
name: cuda-rhel9-x86_64
|
|
description: cuda-rhel9-x86_64
|
|
baseurl: https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64
|
|
gpgkey: https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/D42D0685.pub
|
|
|
|
- name: Enable CRB repo
|
|
ansible.builtin.ini_file:
|
|
path: /etc/yum.repos.d/{{ repo_file }}
|
|
section: "{{ repo_id }}"
|
|
option: enabled
|
|
value: "1"
|
|
no_extra_spaces: true
|
|
vars:
|
|
repo_id: crb
|
|
repo_file: rocky.repo
|
|
|
|
- name: NVidia Driver RPMs
|
|
ansible.builtin.package:
|
|
name:
|
|
- gcc
|
|
- make
|
|
- dkms
|
|
- kernel-devel
|
|
- kernel-headers
|
|
- cuda-toolkit
|
|
# nvidia-smi
|
|
- nvidia-container-toolkit
|
|
state: present
|
|
update_cache: false
|
|
# when: 0 > 1
|
|
|
|
- name: NVidia Kernel Module
|
|
ansible.builtin.dnf:
|
|
name: '@nvidia-driver:580-dkms'
|
|
state: present
|
|
update_cache: false
|
|
# when: 0 > 1
|
|
#
|
|
# Add:
|
|
# dkms autoinstall -k $(uname -r)
|
|
|
|
|
|
# post-install:
|
|
# dkms install nvidia/580.173.02 -k $(uname -r) 2>&1 && sudo modprobe nvidia
|