diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index d782a5f8..238cd77c 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -32,10 +32,8 @@ jobs:
     strategy:
       matrix:
         python-version: [3.7]
-        torch: [1.5.0, 1.6.0, 1.7.0, 1.8.0, 1.9.0]
+        torch: [1.6.0, 1.7.0, 1.8.0, 1.9.0]
         include:
-          - torch: 1.5.0
-            torchvision: 0.6.0
           - torch: 1.6.0
             torchvision: 0.7.0
           - torch: 1.7.0
@@ -44,6 +42,15 @@ jobs:
             torchvision: 0.9.0
           - torch: 1.9.0
             torchvision: 0.10.0
+          - torch: 1.8.0
+            torchvision: 0.9.0
+            python-version: 3.9
+          - torch: 1.9.0
+            torchvision: 0.10.0
+            python-version: 3.8
+          - torch: 1.9.0
+            torchvision: 0.10.0
+            python-version: 3.9
     steps:
       - uses: actions/checkout@v2
       - name: Set up Python ${{ matrix.python-version }}
@@ -81,11 +88,8 @@ jobs:
     strategy:
       matrix:
         python-version: [3.7]
-        torch: [1.5.0+cu101, 1.6.0+cu101, 1.7.0+cu101, 1.8.0+cu101]
+        torch: [1.6.0+cu101, 1.7.0+cu101, 1.8.0+cu101]
         include:
-          - torch: 1.5.0+cu101
-            torch_version: 1.5.0
-            torchvision: 0.6.0+cu101
           - torch: 1.6.0+cu101
             torch_version: 1.6.0
             torchvision: 0.7.0+cu101
@@ -103,6 +107,10 @@ jobs:
             torch_version: 1.8.0
             torchvision: 0.9.0+cu101
             python-version: 3.8
+          - torch: 1.8.0+cu101
+            torch_version: 1.8.0
+            torchvision: 0.9.0+cu101
+            python-version: 3.9
     steps:
       - uses: actions/checkout@v2
       - name: Set up Python ${{ matrix.python-version }}
@@ -178,6 +186,14 @@ jobs:
           - torch: 1.9.0+cu102
             torch_version: 1.9.0
             torchvision: 0.10.0+cu102
+          - torch: 1.9.0+cu102
+            torch_version: 1.9.0
+            torchvision: 0.10.0+cu102
+            python-version: 3.8
+          - torch: 1.9.0+cu102
+            torch_version: 1.9.0
+            torchvision: 0.10.0+cu102
+            python-version: 3.9
     steps:
       - uses: actions/checkout@v2
       - name: Set up Python ${{ matrix.python-version }}
diff --git a/setup.py b/setup.py
index 9ca736b0..c662b482 100644
--- a/setup.py
+++ b/setup.py
@@ -175,6 +175,7 @@ if __name__ == '__main__':
             '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'),