Support python3.9 and pytorch 1.9 in CI (#269)

* support python3.9 and topyrch 1.9 * fix a bug * fix a bug
open-mmlab · Sep 12, 2021 · 1fa9280 · 1fa9280
1 parent e572c51
commit 1fa9280
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -72,7 +72,7 @@ jobs:
     strategy:
       matrix:
         python-version: [3.7]
-        torch: [1.3.1, 1.5.1+cu101, 1.7.1+cu101]
+        torch: [1.3.1, 1.5.1+cu101, 1.7.1+cu101, 1.9.0+cu102]
         include:
           - torch: 1.3.1
             torch_version: torch1.3.0
@@ -83,14 +83,21 @@ jobs:
           - torch: 1.7.1+cu101
             torch_version: torch1.7.0
             torchvision: 0.8.2+cu101
-          - torch: 1.7.1+cu101
-            torch_version: torch1.7.0
-            torchvision: 0.8.2+cu101
+          - torch: 1.9.0+cu102
+            torch_version: torch1.9.0
+            torchvision: 0.10.0+cu102
+          - torch: 1.9.0+cu102
+            torch_version: torch1.9.0
+            torchvision: 0.10.0+cu102
             python-version: 3.6
-          - torch: 1.7.1+cu101
-            torch_version: torch1.7.0
-            torchvision: 0.8.2+cu101
+          - torch: 1.9.0+cu102
+            torch_version: torch1.9.0
+            torchvision: 0.10.0+cu102
             python-version: 3.8
+          - torch: 1.9.0+cu102
+            torch_version: torch1.9.0
+            torchvision: 0.10.0+cu102
+            python-version: 3.9
 
     steps:
       - uses: actions/checkout@v2

diff --git a/setup.py b/setup.py
@@ -198,6 +198,7 @@ def add_mim_extension():
             'Programming Language :: Python :: 3.6',
             'Programming Language :: Python :: 3.7',
             'Programming Language :: Python :: 3.8',
+            'Programming Language :: Python :: 3.9',
         ],
         license='Apache License 2.0',
         setup_requires=parse_requirements('requirements/build.txt'),