diff --git a/.github/workflows/cuda.yml b/.github/workflows/cuda.yml
index 3e2bf218a0..476ff69a48 100644
--- a/.github/workflows/cuda.yml
+++ b/.github/workflows/cuda.yml
@@ -11,7 +11,7 @@ concurrency:
 jobs:
   test:
     name: Test on CUDA Build
-    runs-on: nvidia
+    runs-on: gpu
     if: github.repository_owner == 'deepmodeling'
     container:
       image: ghcr.io/deepmodeling/abacus-cuda