Update wheels.yml

faberno · Oct 4, 2024 · 747032e · 747032e
1 parent 35b87d3
commit 747032e
Showing 1 changed file with 16 additions and 8 deletions.
diff --git 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