test

open-mmlab · Feb 27, 2025 · 6bdd549 · 6bdd549
1 parent 4158852
commit 6bdd549
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/pr_stage_test.yml b/.github/workflows/pr_stage_test.yml
@@ -68,8 +68,9 @@ jobs:
       MKL_THREADING_LAYER: GNU
     strategy:
       matrix:
-        python-version: ['3.9','3.10','3.11']
+        python-version: ['3.9','3.10']
         torch: ['2.0.0','2.3.1','2.5.1']
+        cuda: ['cu118','cu124']
     steps:
       - name: Check out repo
         uses: actions/checkout@v3
@@ -87,7 +88,7 @@ jobs:
       - name: Install dependencies
         run: |
           apt-get update && apt-get install -y ffmpeg libsm6 libxext6 git ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6
-          python -m pip install torch==${{matrix.torch}} --index-url https://download.pytorch.org/whl/cu118
+          python -m pip install torch==${{matrix.torch}} --index-url https://download.pytorch.org/whl/${{matrix.cuda}}
           python -m pip install -e . -v
           python -m pip install -r requirements/tests.txt
           python -m pip install openmim