Funk-lab/ansible/roles/rocm/tasks/main.yml

60 lines
1.5 KiB
YAML

---
- name: Import AMD GPG key
ansible.builtin.rpm_key:
key: "{{ rocm_gpg_key }}"
state: present
- name: Add amdgpu repository
ansible.builtin.yum_repository:
name: amdgpu
description: AMD GPU drivers
baseurl: "{{ rocm_amdgpu_repo }}"
enabled: true
gpgcheck: true
gpgkey: "{{ rocm_gpg_key }}"
- name: Add ROCm repository
ansible.builtin.yum_repository:
name: rocm
description: ROCm
baseurl: "{{ rocm_repo }}"
enabled: true
gpgcheck: true
gpgkey: "{{ rocm_gpg_key }}"
- name: Install kernel headers and devel (requis pour amdgpu-dkms)
ansible.builtin.dnf:
name:
- kernel-headers
- kernel-devel
state: present
- name: Install ROCm packages
ansible.builtin.dnf:
name: "{{ rocm_packages }}"
state: present
notify: Reload amdgpu module
- name: Add ansible user to video and render groups
ansible.builtin.user:
name: "{{ ansible_service_user }}"
groups:
- video
- render
append: true
- name: Set HSA_OVERRIDE_GFX_VERSION system-wide
ansible.builtin.lineinfile:
path: /etc/environment
regexp: '^HSA_OVERRIDE_GFX_VERSION='
line: "HSA_OVERRIDE_GFX_VERSION={{ hsa_override_gfx_version }}"
create: true
mode: '0644'
- name: Set HSA_OVERRIDE_GFX_VERSION dans profile.d (sessions interactives)
ansible.builtin.copy:
dest: /etc/profile.d/rocm.sh
content: |
export HSA_OVERRIDE_GFX_VERSION={{ hsa_override_gfx_version }}
export PATH=$PATH:/opt/rocm/bin
mode: '0644'