See if we can install cuda-toolkit

jloveric · Sep 9, 2023 · 5bafc3b · 5bafc3b
1 parent 7a78c41
commit 5bafc3b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.github/workflows/python-app.yml b/.github/workflows/python-app.yml
@@ -11,10 +11,14 @@ jobs:
       fail-fast: false
       matrix:
         python-version: ["3.10"]
-        poetry-version: [1.3.1]
+        poetry-version: [1.6.0]
         os: [ubuntu-latest, macos-latest] #[ubuntu-latest,macos-latest, windows-latest] windows and mac are flaky
     runs-on: ${{ matrix.os }}
     steps:
+      - uses: Jimver/cuda-toolkit@v0.2.11
+        id: cuda-toolkit
+        with:
+          cuda: '12.1.0'
       - uses: actions/checkout@v2
       - uses: actions/setup-python@v2
         with: