diff --git a/.github/workflows/python-app.yml b/.github/workflows/python-app.yml
index 1cbddbe..1cb7016 100644
--- 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: