enable_gpu_nodepool

Humanitec-DemoOrg · Sep 20, 2023 · c2128ee · c2128ee
1 parent d251f46
commit c2128ee
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/resources/terraform/azure-aks-new/aks-gpu.tf b/resources/terraform/azure-aks-new/aks-gpu.tf
@@ -0,0 +1,12 @@
+# https://registry.terraform.io/providers/hashicorp/azurerm/latest/docs/resources/kubernetes_cluster_node_pool
+resource "azurerm_kubernetes_cluster_node_pool" "gpu" {
+  count                 = var.enable_gpu_nodepool ? 1 : 0
+  name                  = "gpu"
+  kubernetes_cluster_id = azurerm_kubernetes_cluster.aks.id
+  vm_size               = "Standard_NC4as_T4_v3"
+  node_count            = 1
+
+  node_taints = [
+    "sku=gpu:NoSchedule",
+  ]
+}
diff --git a/resources/terraform/azure-aks-new/variables.tf b/resources/terraform/azure-aks-new/variables.tf
@@ -28,4 +28,10 @@ variable "node_size" {
 variable "existing_acr_name" {
   description = "Name of the existing ACR to attach to AKS"
   type        = string
+}
+
+variable "enable_gpu_nodepool" {
+  description = "Create a nodepool dedicated to GPU."
+  type        = bool
+  default     = false
 }