From 9188be39c6d8cf07002e84e168c52f56480ab52f Mon Sep 17 00:00:00 2001
From: Story Crater Bot <19826264+Riotpiaole@users.noreply.github.com>
Date: Tue, 11 Aug 2026 10:31:22 -0700
Subject: [PATCH] fix(terraform): correct NVIDIA extension names to
 nonfree-kmod-nvidia-production/nvidia-container-toolkit-production, add
 required nvidia kernel modules to worker config

---
 terraform/talos_schematic.tf     | 9 ++++++---
 terraform/templates/worker.tftpl | 8 ++++++++
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/terraform/talos_schematic.tf b/terraform/talos_schematic.tf
index 815e4ab..1061e02 100644
--- a/terraform/talos_schematic.tf
+++ b/terraform/talos_schematic.tf
@@ -19,14 +19,17 @@ data "talos_image_factory_urls" "longhorn_installer" {
   platform      = "metal"
 }
 
-# GPU-enabled schematic with NVIDIA drivers and toolkit
+# GPU-enabled schematic with NVIDIA drivers and toolkit (+ Longhorn deps, since
+# worker nodes also run the Longhorn CSI DaemonSet)
 resource "talos_image_factory_schematic" "gpu_enabled" {
   schematic = jsonencode({
     customization = {
       systemExtensions = {
         officialExtensions = [
-          "siderolabs/nvidia-driver",
-          "siderolabs/nvidia-container-toolkit",
+          "siderolabs/iscsi-tools",
+          "siderolabs/util-linux-tools",
+          "siderolabs/nonfree-kmod-nvidia-production",
+          "siderolabs/nvidia-container-toolkit-production",
         ]
       }
     }
diff --git a/terraform/templates/worker.tftpl b/terraform/templates/worker.tftpl
index 5709a5f..507ab71 100644
--- a/terraform/templates/worker.tftpl
+++ b/terraform/templates/worker.tftpl
@@ -48,6 +48,14 @@ machine:
     image: ${factory_image}
     wipe: true
     grubUseUKICmdline: true
+%{ if gpu_count > 0 ~}
+  kernel:
+    modules:
+      - name: nvidia
+      - name: nvidia_uvm
+      - name: nvidia_drm
+      - name: nvidia_modeset
+%{ endif ~}
   disks:
 %{ for disk in extra_disks ~}
     - device: ${disk.device}