update

Author: lorenzotomada

.github/workflows/tests.yml

@@ -23,6 +23,13 @@ jobs:
     steps:
     - uses: actions/checkout@v3
 
+    - name: Set CUDA environment variables
+      run: |
+        echo "export LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:\$LD_LIBRARY_PATH" >> ~/.bashrc
+        echo "export CUDA_HOME=/usr/local/cuda" >> ~/.bashrc
+        source ~/.bashrc
+      shell: bash -e {0}
+
     - name: Install dependencies
       run: |
         /opt/environments/python/jupyter/bin/python -m pip install --upgrade pip