Use docker image

Signed-off-by: SimJeg <[email protected]>

Author: SimJeg

.github/workflows/test.yml

@@ -9,14 +9,16 @@ on:
 jobs:
   test:
     runs-on: linux-amd64-gpu-l4-latest-1
+    container:
+      image: nvcr.io/nvidia/pytorch:25.10-py3
     steps:
       - uses: actions/checkout@v3
 
       - name: Verify environment
         run: |
           nvidia-smi
+          nvcc --version
           python3 --version
-          echo "CUDA_HOME=/usr/local/cuda" >> $GITHUB_ENV
       - name: Install uv
         uses: astral-sh/setup-uv@v6
         with: