[Fix] CI min gpu version and GitHub merge test (#2782)

open-mmlab · Oct 19, 2023 · 12b595c · 12b595c
1 parent 5a533e7
commit 12b595c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/.circleci/test.yml b/.circleci/test.yml
@@ -85,10 +85,10 @@ jobs:
         type: string
       cuda:
         type: enum
-        enum: ["10.1", "10.2", "11.1", "11.7"]
+        enum: ["11.1", "11.7"]
       cudnn:
         type: integer
-        default: 7
+        default: 8
     machine:
       image: ubuntu-2004-cuda-11.4:202110-01
       # docker_layer_caching: true
@@ -167,7 +167,7 @@ workflows:
           torch: 1.8.1
           # Use double quotation mark to explicitly specify its type
           # as string instead of number
-          cuda: "10.2"
+          cuda: "11.1"
           requires:
             - hold
       - build_cuda:
@@ -184,7 +184,7 @@ workflows:
       - build_cuda:
           name: minimum_version_gpu
           torch: 1.8.1
-          cuda: "10.2"
+          cuda: "11.1"
           filters:
             branches:
               only:

diff --git a/.github/workflows/merge_stage_test.yml b/.github/workflows/merge_stage_test.yml
@@ -59,7 +59,7 @@ jobs:
     strategy:
       matrix:
         python-version: [3.7]
-        torch: [1.8.1, 1.9.1, 1.10.1, 1.11.0, 1.12.0, 1.13.0, 2.0.0]
+        torch: [1.8.1, 1.9.1, 1.10.1, 1.11.0, 1.12.0, 1.13.0]
         include:
           - torch: 1.8.1
             torchvision: 0.9.1
@@ -73,9 +73,9 @@ jobs:
             torchvision: 0.13.0
           - torch: 1.13.0
             torchvision: 0.14.0
-          - torch: 2.0.0
+          - python-version: 3.8
+            torch: 2.0.0
             torchvision: 0.15.1
-            python-version: 3.8
     steps:
       - uses: actions/checkout@v3
       - name: Set up Python ${{ matrix.python-version }}