Hot fix for gpu build wheel pipeline

BabitMF · Dec 13, 2023 · f5c0f2b · f5c0f2b
1 parent c4fb7b7
commit f5c0f2b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/build_wheels.yml b/.github/workflows/build_wheels.yml
@@ -69,6 +69,7 @@ jobs:
           CMAKE_BUILD_PARALLEL_LEVEL=${{ steps.set_env.outputs.count }}
           GIT_SHA=${{ github.sha }}
           DEVICE=${{ matrix.device }}
+          CUDA_PATH=/usr/local/cuda
           SCRIPT_EXEC_MODE=x86
         CIBW_BUILD_VERBOSITY: 1
         CIBW_BUILD: ${{ matrix.py_ver_prefix }}-*