From 1ed20ca0831ed2bdb56200134369f53d770ed68e Mon Sep 17 00:00:00 2001
From: Kentechx <kd_shen@163.com>
Date: Wed, 26 Jul 2023 01:47:03 +0800
Subject: [PATCH] update workflow

---
 .github/workflows/python-publish.yml | 4 ++++
 1 file changed, 4 insertions(+)

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