update workflow

kentechx · Jul 25, 2023 · 8c7a7da · 8c7a7da
1 parent 1ed20ca
commit 8c7a7da
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
@@ -18,8 +18,16 @@ jobs:
   deploy:
     runs-on: ubuntu-latest
     steps:
-      - name: cuda-toolkit
-        uses: Jimver/[email protected]
+      - uses: Jimver/[email protected]
+        id: cuda-toolkit
+        with:
+          cuda: '11.8.0'
+          method: 'network'
+          sub-packages: '["nvcc"]'
+        run: |
+          echo "Installed cuda version is: ${{steps.cuda-toolkit.outputs.cuda}}"
+          echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
+          nvcc -V
       - uses: actions/checkout@v3
       - name: Set up Python
         uses: actions/setup-python@v4