set cuda env in test workflow

Signed-off-by: Max Jeblick <maximilianjeblick@gmail.com>

Author: maxjeblick

.github/workflows/test.yml

@@ -16,6 +16,14 @@ jobs:
         with:
           python-version: 3.10.11
 
+      - name: Setup CUDA
+        uses: Jimver/cuda-toolkit@v0.2.11
+        with:
+          cuda: '12.1.0'
+
+      - name: Set CUDA_HOME
+        run: echo "CUDA_HOME=/usr/local/cuda" >> $GITHUB_ENV
+
       - name: Install uv
         uses: astral-sh/setup-uv@v6
         with: