diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 736f36d..507c969 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -15,6 +15,10 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v4
+      - uses: Jimver/cuda-toolkit@v0.2.24
+        id: cuda-toolkit
+        with:
+          cuda: '12.5.0'
       - name: Setup Python virtual environment
         shell: bash
         run: |