From 061044c2fb30f5aae80a558ede763005c3b3f25e Mon Sep 17 00:00:00 2001
From: geoffsee <>
Date: Sun, 22 Jun 2025 08:27:57 -0400
Subject: [PATCH] --amend

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

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: |