fix(terraform): correct NVIDIA extension names to nonfree-kmod-nvidia-production/nvidia-container-toolkit-production, add required nvidia kernel modules to worker config

2026-08-11 10:31:22 -07:00
parent d4508afc07
commit 9188be39c6
2 changed files with 14 additions and 3 deletions
@@ -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}