Add GPU nodegroup

yuvipanda · Mar 25, 2024 · 63d3af2 · 63d3af2
1 parent 51581bc
commit 63d3af2
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/eksctl/bican.jsonnet b/eksctl/bican.jsonnet
@@ -28,6 +28,15 @@ local notebookNodes = [
     { instanceType: "r5.xlarge" },
     { instanceType: "r5.4xlarge" },
     { instanceType: "r5.16xlarge" },
+    {
+      instanceType: "g4dn.xlarge",
+      tags+: {
+        "k8s.io/cluster-autoscaler/node-template/resources/nvidia.com/gpu": "1"
+      },
+      // Allow provisioning GPUs across all AZs, to prevent situation where all
+      // GPUs in a single AZ are in use and no new nodes can be spawned
+      availabilityZones: masterAzs,
+    },
 ];
 local daskNodes = [
     // Node definitions for dask worker nodes. Config here is merged