diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
index f2e311c..1e2871f 100644
--- a/.github/workflows/python-package.yml
+++ b/.github/workflows/python-package.yml
@@ -26,7 +26,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install pytest torch==1.10.0
+        python -m pip install pytest torch==2.0.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
     - name: Test with pytest
       run: |
         python setup.py test