Adding tasks to essi/configure.yml to make eessi configure

gpu node automatically

Author: claudia-lola

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

@@ -294,6 +294,22 @@
           ansible.builtin.command:
             cmd: "cvmfs_config setup"
 
+        # configure gpus
+        - name: Check for NVIDIA driver
+          ansible.builtin.stat:
+            path: /dev/nvidia0
+          register: nvidia_driver
+
+        - name: Set fact if NVIDIA driver 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
+
     - name: Configure VGPUs
       ansible.builtin.include_role:
         name: stackhpc.linux.vgpu

ansible/roles/eessi/tasks/configure.yml

@@ -15,3 +15,19 @@
 - name: Ensure CVMFS config is setup # noqa: no-changed-when
   ansible.builtin.command:
     cmd: "cvmfs_config setup"
+
+# configure gpus
+- name: Check for NVIDIA driver
+  ansible.builtin.stat:
+    path: /dev/nvidia0
+  register: nvidia_driver
+
+- name: Set fact if NVIDIA driver 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