[Fix] Fix CI for torch2.1.0

open-mmlab · Nov 2, 2023 · dc4d48a · dc4d48a
1 parent 5b0e0ce
commit dc4d48a
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 10 deletions.
diff --git a/.github/workflows/merge_stage_test.yml b/.github/workflows/merge_stage_test.yml
@@ -286,10 +286,6 @@ jobs:
         torch: [1.8.1]
         torchvision: [0.9.1]
         include:
-          - python-version: 3.8
-            platform: cu117
-            torch: 2.0.0
-            torchvision: 0.15.1
           - python-version: 3.8
             platform: cu118
             torch: 2.1.0

diff --git a/.github/workflows/pr_stage_test.yml b/.github/workflows/pr_stage_test.yml
@@ -137,14 +137,10 @@ jobs:
     strategy:
       matrix:
         python-version: [3.7]
-        platform: [cpu]
+        platform: [cpu, cu111]
         torch: [1.8.1]
         torchvision: [0.9.1]
         include:
-          - python-version: 3.8
-            platform: cu117
-            torch: 2.0.0
-            torchvision: 0.15.1
           - python-version: 3.8
             platform: cu118
             torch: 2.1.0

diff --git a/requirements/tests.txt b/requirements/tests.txt
@@ -7,7 +7,6 @@ dvclive
 lion-pytorch
 lmdb
 mlflow
-neptune
 parameterized
 pydantic==1.10.9
 pytest