diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 64421da8e..d90771db0 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -14,7 +14,7 @@ jobs:
     strategy:
         matrix:
           os: [ubuntu-16.04, ubuntu-18.04, windows-latest, macos-latest]
-          python-version: [3.6, 3.7]
+          python-version: [3.6, 3.7, 3.8, 3.9]
 
     steps:
     - uses: actions/checkout@v2
@@ -39,7 +39,7 @@ jobs:
     - name: Install torch on Windows
       if: runner.os == 'Windows'
       run: |
-        pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
+        pip install torch===1.7.1 torchvision===0.8.2 -f https://download.pytorch.org/whl/torch_stable.html
 
     - name: Install test dependencies
       run: |
diff --git a/setup.py b/setup.py
index c2849e6c0..88763dee4 100644
--- a/setup.py
+++ b/setup.py
@@ -119,8 +119,10 @@
         "Programming Language :: Python :: 3",
         "Programming Language :: Python :: 3.6",
         "Programming Language :: Python :: 3.7",
+        "Programming Language :: Python :: 3.8",
+        "Programming Language :: Python :: 3.9",
         "Topic :: Scientific/Engineering :: Artificial Intelligence"],
-    python_requires=">=3.6,<3.8",
+    python_requires=">=3.6",
     setup_requires=[
         "numpy<1.20.0",
     ],