update workflow

kentechx · Jul 25, 2023 · 1ed20ca · 1ed20ca
1 parent 2e86f2c
commit 1ed20ca
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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/[email protected]
       - 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