Pin AMZN2023 images for non-gpu instances, starting amzn2023 migration (

#5400)
pytorch · Jul 11, 2024 · 4e860d3 · 4e860d3
1 parent 22b1add
commit 4e860d3
Showing 1 changed file with 17 additions and 7 deletions.
diff --git a/.github/scale-config.yml b/.github/scale-config.yml
@@ -27,60 +27,68 @@ runner_types:
     is_ephemeral: false
     max_available: 1000
     os: linux
+    ami: al2023-ami-2023.5.20240701.0-kernel-6.1-x86_64
   linux.10xlarge.avx2:
     disk_size: 200
     instance_type: m4.10xlarge
     is_ephemeral: false
     max_available: 60
     os: linux
+    ami: al2023-ami-2023.5.20240701.0-kernel-6.1-x86_64
   linux.24xl.spr-metal:
     disk_size: 200
     instance_type: c7i.metal-24xl
     is_ephemeral: false
     max_available: 150
     os: linux
+    ami: al2023-ami-2023.5.20240701.0-kernel-6.1-x86_64
   linux.16xlarge.spr:
     disk_size: 200
     instance_type: c7i.16xlarge
     is_ephemeral: false
     max_available: 150
     os: linux
+    ami: al2023-ami-2023.5.20240701.0-kernel-6.1-x86_64
   linux.12xlarge.ephemeral:
     disk_size: 200
     instance_type: c5.12xlarge
     is_ephemeral: true
     max_available: 300
     os: linux
-  linux.16xlarge.nvidia.gpu:
-    disk_size: 150
-    instance_type: g3.16xlarge
-    is_ephemeral: false
-    max_available: 150
-    os: linux
+    ami: al2023-ami-2023.5.20240701.0-kernel-6.1-x86_64
   linux.24xlarge:
     disk_size: 150
     instance_type: c5.24xlarge
     is_ephemeral: false
     max_available: 250
     os: linux
+    ami: al2023-ami-2023.5.20240701.0-kernel-6.1-x86_64
   linux.2xlarge:
     disk_size: 150
     instance_type: c5.2xlarge
     is_ephemeral: false
     max_available: 3120
     os: linux
+    ami: al2023-ami-2023.5.20240701.0-kernel-6.1-x86_64
   linux.4xlarge:
     disk_size: 150
     instance_type: c5.4xlarge
     is_ephemeral: false
     max_available: 1000
     os: linux
+    ami: al2023-ami-2023.5.20240701.0-kernel-6.1-x86_64
   linux.4xlarge.nvidia.gpu:
     disk_size: 150
     instance_type: g3.4xlarge
     is_ephemeral: false
     max_available: 520
     os: linux
+  linux.16xlarge.nvidia.gpu:
+    disk_size: 150
+    instance_type: g3.16xlarge
+    is_ephemeral: false
+    max_available: 150
+    os: linux
   linux.8xlarge.nvidia.gpu:
     disk_size: 150
     instance_type: g3.8xlarge
@@ -122,13 +130,15 @@ runner_types:
     instance_type: g6.4xlarge
     is_ephemeral: false
     max_available: 30
-    os: linux    
+    os: linux
+    ami: al2023-ami-2023.5.20240701.0-kernel-6.1-x86_64
   linux.large:
     max_available: 1200
     disk_size: 15
     instance_type: c5.large
     is_ephemeral: false
     os: linux
+    ami: al2023-ami-2023.5.20240701.0-kernel-6.1-x86_64
   linux.arm64.2xlarge:
     disk_size: 256
     instance_type: t4g.2xlarge