2i2c-aws-us, researchdelight: revert remove gpu image choices

2i2c-org · May 29, 2023 · 4aed868 · 4aed868
1 parent 710bc23
commit 4aed868
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/config/clusters/2i2c-aws-us/researchdelight.values.yaml b/config/clusters/2i2c-aws-us/researchdelight.values.yaml
@@ -232,6 +232,21 @@ basehub:
         - display_name: NVIDIA Tesla T4, ~16 GB, ~4 CPUs
           slug: gpu
           description: "Start a container on a dedicated node with a GPU"
+          profile_options:
+            image:
+              display_name: Image
+              choices:
+                tensorflow:
+                  display_name: Pangeo Tensorflow ML Notebook
+                  slug: "tensorflow"
+                  kubespawner_override:
+                    image: "pangeo/ml-notebook:b9584f6"
+                pytorch:
+                  display_name: Pangeo PyTorch ML Notebook
+                  default: true
+                  slug: "pytorch"
+                  kubespawner_override:
+                    image: "pangeo/pytorch-notebook:b9584f6"
           kubespawner_override:
             mem_limit: null
             mem_guarantee: 14G