Update wheels_build.yml

chengzeyi · Feb 9, 2024 · 0e32534 · 0e32534
1 parent d85e16b
commit 0e32534
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.github/workflows/wheels_build.yml b/.github/workflows/wheels_build.yml
@@ -143,7 +143,9 @@ jobs:
           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 --toolkitpath=/usr/local/cuda --no-opengl-libs --no-man-page --no-drm &&
-          rm ./cuda.run
+          rm ./cuda.run &&
+          echo "CUDA_HOME=/usr/local/cuda" >> ${GITHUB_ENV} &&
+          echo "PATH=/usr/local/cuda/bin:$PATH" >> ${GITHUB_ENV}
 
       - if: runner.os == 'Linux'
         name: (Linux) install python