Skip to content

Commit bfa2e53

Browse files
committed
fix(google_gke): Add variable for GPU driver version
1 parent 85787df commit bfa2e53

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

google_gke/cluster.tf

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -288,6 +288,10 @@ resource "google_container_node_pool" "pools" {
288288
content {
289289
type = local.node_pools_guest_accelerator[each.key].type
290290
count = local.node_pools_guest_accelerator[each.key].count
291+
292+
gpu_driver_installation_config {
293+
gpu_driver_version = local.node_pools_guest_accelerator[each.key].gpu_driver_version
294+
}
291295
}
292296
}
293297

google_gke/variables.tf

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -267,7 +267,11 @@ variable "node_pools_taints" {
267267

268268
variable "node_pools_guest_accelerator" {
269269
description = "Map containing node pools guest accelerator. Each node pool's name is the key. See locals.tf for defaults."
270-
type = map(map(string))
270+
type = map(object({
271+
type = string
272+
count = number
273+
gpu_driver_version = optional(string, "LATEST")
274+
}))
271275
default = {
272276
tf-default-node-pool = {}
273277
}

0 commit comments

Comments
 (0)