Merge pull request #16 from chiayi/update-machine

Add labels to match ray nodeselector
GoogleCloudPlatform · Aug 14, 2023 · 94dc853 · 94dc853
2 parents da2a6f0 + 9dfa99c
commit 94dc853
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ray-on-gke/platform/modules/gke_standard/main.tf b/ray-on-gke/platform/modules/gke_standard/main.tf
@@ -74,6 +74,11 @@ resource "google_container_node_pool" "gpu_pool" {
       env = var.project_id
     }
 
+    guest_accelerator {
+      type = "nvidia-tesla-t4"
+      count = 2
+    }
+
     # preemptible  = true
     image_type   = "cos_containerd"
     machine_type = "n1-standard-16"