errofix r Fai:led to detect a default CUDA architecture.

pytorch · May 16, 2024 · ce9925d · ce9925d
1 parent 7d16fdf
commit ce9925d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.github/workflows/_build_torch_with_cuda_and_xla.yml b/.github/workflows/_build_torch_with_cuda_and_xla.yml
@@ -32,6 +32,13 @@ jobs:
         shell: bash
         run: |
           echo "${GCLOUD_SERVICE_KEY}" > $GOOGLE_APPLICATION_CREDENTIALS
+      - name: Setup CUDA environment
+        shell: bash
+        run: |
+          echo "PATH=$PATH:/usr/local/cuda-12.1/bin" >> $GITHUB_ENV
+          echo "LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.1/lib64" >> $GITHUB_ENV
+      - name: Check GPU
+        run: nvidia-smi
       - name: Checkout PyTorch Repo
         uses: actions/checkout@v4
         with: