chore: minor

Author: chenzhuofu

.github/runs-on.yml

@@ -1,16 +1,19 @@
 images:
-  ubuntu-22:
-    platform: "linux"
-    arch: "x64"
-    owner: "099720109477" # Official Ubuntu owner
-    name: "ubuntu/images/hvm-ssd/ubuntu-jammy-22.04-amd64-server-*" # Official Ubuntu image
   dlami-x64:
     platform: "linux"
     arch: "x64"
     owner: "898082745236" # AWS
     name: "Deep Learning Base OSS Nvidia Driver GPU AMI (Ubuntu 22.04)*"
+  ubuntu-22:
+    platform: "linux"
+    arch: "x64"
+    owner: "099720109477" # Official Ubuntu owner
+    name: "ubuntu/images/hvm-ssd/ubuntu-jammy-22.04-amd64-server-*" # Official Ubuntu image
 
 runners:
   gpu-nvidia:
+    family: ["g4dn.xlarge"]
+    image: dlami-x64
+  gpu-nvidia-lightweight:
     family: ["g4dn.xlarge"]
     image: ubuntu-22

.github/workflows/gpu-ci.yml

@@ -7,7 +7,7 @@ concurrency:
 jobs:
   gpu-ci:
     name: GPU-CI
-    runs-on: [runs-on,runner=gpu-nvidia]
+    runs-on: [runs-on,runner=gpu-nvidia-lightweight]
 
     strategy:
       max-parallel: 1