stash

2i2c-org · Sep 24, 2024 · f9ffc5b · f9ffc5b
1 parent 5e30140
commit f9ffc5b
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/config/clusters/victor/staging.values.yaml b/config/clusters/victor/staging.values.yaml
@@ -47,24 +47,57 @@ basehub:
             mem_guarantee: 6.5G
             node_selector:
               node.kubernetes.io/instance-type: m5.large
+          profile_options: &profile_options
+            image:
+              display_name: Image
+              choices:
+                victor-notebook:
+                  display_name: Victor Notebook
+                  default: true
+                  kubespawner_override:
+                    image: quay.io/volcanocyber/victor-notebook:a045ad3616d1
+                pytorch-notebook:
+                  display_name: Pangeo ML Notebook (Pytorch)
+                  kubespawner_override:
+                    image: "quay.io/pangeo/pytorch-notebook:2024.09.11"
+                ml-notebook:
+                  display_name: Pangeo ML Notebook (Tensorflow)
+                  kubespawner_override:
+                    image: "quay.io/pangeo/ml-notebook:2024.09.11"
         - display_name: "Medium: m5.xlarge"
           description: "~4 CPU, ~15G RAM"
           kubespawner_override:
             mem_limit: 15G
             mem_guarantee: 12G
             node_selector:
               node.kubernetes.io/instance-type: m5.xlarge
+          profile_options: *profile_options
         - display_name: "Large: m5.2xlarge"
           description: "~8 CPU, ~30G RAM"
           kubespawner_override:
             mem_limit: 30G
             mem_guarantee: 25G
             node_selector:
               node.kubernetes.io/instance-type: m5.2xlarge
+          profile_options: *profile_options
         - display_name: "Huge: m5.8xlarge"
           description: "~16 CPU, ~60G RAM"
           kubespawner_override:
             mem_limit: 60G
             mem_guarantee: 50G
             node_selector:
               node.kubernetes.io/instance-type: m5.8xlarge
+          profile_options: *profile_options
+        - display_name: NVIDIA Tesla T4, ~16 GB, ~4 CPUs
+          description: "Start a container on a dedicated node with a GPU"
+          slug: "gpu"
+          kubespawner_override:
+            environment:
+              NVIDIA_DRIVER_CAPABILITIES: compute,utility
+            mem_limit: null
+            mem_guarantee: 14G
+            node_selector:
+              node.kubernetes.io/instance-type: g4dn.xlarge
+            extra_resource_limits:
+              nvidia.com/gpu: "1"
+          profile_options: *profile_options