2i2c-aws-us, ncar-cisl: add tf/pytorch notebooks for gpu node

2i2c-org · May 29, 2023 · e35dcd1 · e35dcd1
1 parent 4aed868
commit e35dcd1
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/config/clusters/2i2c-aws-us/ncar-cisl.values.yaml b/config/clusters/2i2c-aws-us/ncar-cisl.values.yaml
@@ -237,6 +237,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:2023.05.18"
+                pytorch:
+                  display_name: Pangeo PyTorch ML Notebook
+                  default: true
+                  slug: "pytorch"
+                  kubespawner_override:
+                    image: "pangeo/pytorch-notebook:2023.05.18"
           kubespawner_override:
             mem_limit: null
             mem_guarantee: 14G