Add CUDA to CI

Author: FilipeMaia

.github/workflows/test.yml

@@ -10,6 +10,10 @@ jobs:
   build:
     runs-on: ubuntu-latest
 
+    strategy:
+      matrix:
+        # Define the matrix for whether to install CUDA or not
+        install_cuda: [true, false]
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python
@@ -26,6 +30,11 @@ jobs:
         sudo apt-get update
         sudo apt-get install -y cmake build-essential libhdf5-dev libgsl-dev libfftw3-dev swig libnfft3-dev libtiff-dev libpng-dev 
         pip install pytest setuptools
+    
+    - name: Install NVIDIA CUDA Toolkit (if matrix.install_cuda is true)
+      if: matrix.install_cuda == true
+      run: |
+        sudo apt-get install -y nvidia-cuda-toolkit
 
     - name: Build
       run: |