From 747032e37bc874d9676d13526bbb5e6ae0546419 Mon Sep 17 00:00:00 2001
From: Fabian Schneider <56137202+faberno@users.noreply.github.com>
Date: Fri, 4 Oct 2024 11:17:16 +0200
Subject: [PATCH] Update wheels.yml

---
 .github/workflows/wheels.yml | 24 ++++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
index db3ab7c..3b5d8a9 100644
--- a/.github/workflows/wheels.yml
+++ b/.github/workflows/wheels.yml
@@ -51,19 +51,27 @@ jobs:
     - name: Install CUDA (Linux only)
       if: ${{ runner.os == 'Linux' }}
       run: |
-        echo "CUDA PATH: $CUDA_HOME"
+        CUDA_PATH=${{steps.cuda-toolkit.outputs.CUDA_PATH}}
+        echo "CUDA_PATH=${CUDA_PATH}"
+        export CUDA_PATH=${CUDA_PATH}
+        export PATH="$CUDA_PATH/bin:$PATH"
+        export LD_LIBRARY_PATH="$CUDA_PATH/lib:$LD_LIBRARY_PATH"
+        export LD_LIBRARY_PATH="$CUDA_PATH/lib64:$LD_LIBRARY_PATH"
+        export CUDA_HOME=${CUDA_PATH}
+        
+        echo "CUDA HOME: $CUDA_HOME"
         echo "LD LIBRARY $LD_LIBRARY_PATH"
         echo "PATH $PATH"
 
         echo Readlink before $(readlink /usr/local/cuda)
-        sudo ln -s ${{steps.cuda-toolkit.outputs.CUDA_PATH}} /usr/local/cuda
+        sudo ln -s $CUDA_PATH /usr/local/cuda
         echo Readlink after $(readlink /usr/local/cuda)
-        
-        export CUDA_HOME=/usr/local/cuda
-        export PATH=/usr/local/cuda/bin:$PATH
-        export LD_LIBRARY_PATH=/usr/local/cuda/bin:$LD_LIBRARY_PATH
-        export CUDA_BIN_PATH=${{steps.cuda-toolkit.outputs.CUDA_PATH}}
-        
+    
+        echo "Adding CUDA to CUDA_PATH, PATH and LD_LIBRARY_PATH"
+        echo "CUDA_PATH=${CUDA_PATH}" >> $GITHUB_ENV
+        echo "${CUDA_PATH}/bin" >> $GITHUB_PATH
+        echo "LD_LIBRARY_PATH=${CUDA_PATH}/lib:${LD_LIBRARY_PATH}" >> $GITHUB_ENV
+        echo "LD_LIBRARY_PATH=${CUDA_PATH}/lib64:${LD_LIBRARY_PATH}" >> $GITHUB_ENV
         echo "/usr/local/cuda/bin" >> $GITHUB_PATH
         nvcc -V