Update wheels_build.yml

chengzeyi · Feb 9, 2024 · 91b1f81 · 91b1f81
1 parent 16be916
commit 91b1f81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/wheels_build.yml b/.github/workflows/wheels_build.yml
@@ -142,7 +142,7 @@ jobs:
           sudo apt update && sudo apt install -y wget git rename &&
           sudo apt clean -y && sudo apt autoremove -y &&
           wget -q "${{ steps.cuda_info.outputs.CUDA_INSTALL_SCRIPT }}" -O cuda.run &&
-          sh ./cuda.run --silent --toolkit --no-opengl-libs --no-man-page --no-drm &&
+          sh ./cuda.run --silent --toolkit --toolkitpath=/usr/local/cuda --no-opengl-libs --no-man-page --no-drm &&
           rm ./cuda.run
 
       - if: runner.os == 'Linux'