Set CUDA_HOME

Signed-off-by: SimJeg <sjegou@nvidia.com>

Author: SimJeg

.github/workflows/test.yml

@@ -5,7 +5,6 @@ on:
     branches:
       - main
       - "pull-request/[0-9]+"
-  workflow_dispatch:  # Allows manual triggering from any branch
 
 jobs:
   test:
@@ -17,6 +16,7 @@ jobs:
         run: |
           nvidia-smi
           python3 --version
+          echo "CUDA_HOME=/usr/local/cuda" >> $GITHUB_ENV
       - name: Install uv
         uses: astral-sh/setup-uv@v6
         with: