--- a/package_lists.py	Wed Feb 14 18:45:31 2024 -0800
+++ b/package_lists.py	Wed Feb 14 18:48:08 2024 -0800
@@ -276,42 +276,32 @@
 
 xorg = [
     'kbd',
-
-    # squib GT640 needs driver 470
-    # plus is on 535
-    # dash is on 525
-    # slash is on 525
-
-    # 'libnvidia-cfg1-525',
-    # 'libnvidia-common-525',
-    # # not  on slash ,maybe needed for dash/steam?
-    # #    'libnvidia-compute-525:i386',
-    # 'libnvidia-compute-525',
-    # #'libnvidia-decode-525:i386',
-    # 'libnvidia-decode-525',
-    # #'libnvidia-encode-525:i386',
-    # 'libnvidia-encode-525',
-    # 'libnvidia-extra-525',
-    # #'libnvidia-fbc1-525:i386',
-    # 'libnvidia-fbc1-525',
-    # #'libnvidia-gl-525:i386',
-    # 'libnvidia-gl-525',
-    # 'nvidia-compute-utils-525',
-    # 'nvidia-dkms-525',
-    # 'nvidia-driver-525',
-    # 'nvidia-kernel-common-525',
-    # 'nvidia-kernel-source-525',
-    # 'nvidia-utils-525',
-    # 'xserver-xorg-video-nvidia-525',
     'nvidia-modprobe',
     'nvidia-prime',
     'nvidia-settings',
     'screen-resolution-extra',
     'xserver-xorg',
 ]
-k8s_node_with_nvidia_gpu = [
-    'nvidia-container-toolkit',
-    'nvidia-container-runtime',
-    'cuda-drivers-fabricmanager-535',
-    'nvidia-headless-535-server',
-    ]
+
+
+def k8s_node_with_nvidia_gpu(hostName):
+    version = {
+        'dash': '545',
+        'dot': '545',
+        'slash': '525',
+        'ditto': '535-server',
+        # 'bang': '390-server',  # no longer in ubuntu
+        'squib': '470',
+    }[hostName]
+    number = version.replace('-server', '')
+    return [
+        'nvidia-container-runtime',
+        f'nvidia-headless-{version}',
+        f'nvidia-utils-{version}',  # this gets mysteriously reverted on ditto- see workaround in packages.py
+        f'libnvidia-encode-{number}',
+        f'nvidia-driver-{version}',
+    ] + ([] if 'server' in version else [
+        f'xserver-xorg-video-nvidia-{version}',
+    ])
+
+