diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
index bb23d63..c353a23 100644
--- a/.github/workflows/python-publish.yml
+++ b/.github/workflows/python-publish.yml
@@ -18,6 +18,8 @@ jobs:
   deploy:
     runs-on: ubuntu-latest
     steps:
+      - name: cuda-toolkit
+        uses: Jimver/cuda-toolkit@v0.2.11
       - uses: actions/checkout@v3
       - name: Set up Python
         uses: actions/setup-python@v4
@@ -28,6 +30,8 @@ jobs:
           python -m pip install --upgrade pip
           pip install build
           pip install torch==2.0.0
+          python -c "import torch; print('PyTorch:', torch.__version__)"
+          python -c "import torch; print('CUDA:', torch.version.cuda)"
       - name: Build package
         run: python setup.py sdist
       - name: Publish package