Make eessi configure gpu node automatically

ansible/roles/compute_init/files/compute-init.yml

@@ -277,22 +277,11 @@
         name: basic_users
       when: enable_basic_users
 
-    - name: EESSI
-      when: enable_eessi
       # NB: don't need conditional block on enable_compute as have already exited
       # if not the case
-      block:
-        - name: Copy cvmfs config
-          ansible.builtin.copy:
-            src: /var/tmp/cluster/cvmfs/default.local
-            dest: /etc/cvmfs/default.local
-            owner: root
-            group: root
-            mode: "0644"
-
-        - name: Ensure CVMFS config is setup # noqa: no-changed-when
-          ansible.builtin.command:
-            cmd: "cvmfs_config setup"
+    - name: Configure EESSI
+      ansible.builtin.include_tasks: tasks/eessi.yml
+      when: enable_eessi
 
     - name: Configure VGPUs
       ansible.builtin.include_role:

ansible/roles/compute_init/tasks/install.yml

@@ -54,6 +54,8 @@
       dest: roles/
     - src: ../../nhc
       dest: roles/
+    - src: ../../eessi/tasks/configure.yml
+      dest: tasks/eessi.yml
 
 - name: Add filter_plugins to ansible.cfg
   ansible.builtin.lineinfile:

ansible/roles/eessi/tasks/configure.yml

@@ -15,3 +15,20 @@
 - name: Ensure CVMFS config is setup # noqa: no-changed-when
   ansible.builtin.command:
     cmd: "cvmfs_config setup"
+
+# configure gpus
+- name: Check for NVIDIA GPU
+  ansible.builtin.stat:
+    path: /dev/nvidia0
+  register: nvidia_driver
+
+- name: Set fact if NVIDIA GPU is present
+  ansible.builtin.set_fact:
+    has_nvidia_driver: "{{ nvidia_driver.stat.exists | default(false) }}"
+
+- name: Expose GPU drivers
+  ansible.builtin.shell: |
+    source /cvmfs/software.eessi.io/versions/2023.06/init/bash
+    /cvmfs/software.eessi.io/versions/2023.06/scripts/gpu_support/nvidia/link_nvidia_host_libraries.sh
+  when: has_nvidia_driver
+  changed_when: true