diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index dae8ee0..bb23d63 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -27,6 +27,7 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install build
+          pip install torch==2.0.0
       - name: Build package
         run: python setup.py sdist
       - name: Publish package