Use ML images with correct CUDA versions

Brings in pangeo-data/pangeo-docker-images#389 Based on pangeo-data/pangeo-docker-images#390, start making T4 the default. Folks can still use K80 if they want. This makes it easier to use CUDA based GPU accelerated code. Follow-up to 2i2c-org#1766
yuvipanda · Oct 14, 2022 · cb1827d · cb1827d
1 parent 5d22425
commit cb1827d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/config/clusters/leap/common.values.yaml b/config/clusters/leap/common.values.yaml
@@ -117,7 +117,6 @@ basehub:
               choices:
                 k80:
                   display_name: NVidia Tesla K80
-                  default: true
                   slug: k80
                   kubespawner_override:
                     node_selector:
@@ -126,6 +125,7 @@ basehub:
                 t4:
                   display_name: NVidia Tesla T4
                   slug: t4
+                  default: true
                   kubespawner_override:
                     node_selector:
                       node.kubernetes.io/instance-type: n1-standard-8
@@ -138,12 +138,12 @@ basehub:
                   default: true
                   slug: "tensorflow"
                   kubespawner_override:
-                    image: "pangeo/ml-notebook:f84546a"
+                    image: "pangeo/ml-notebook:2022.10.13"
                 pytorch:
                   display_name: Pangeo PyTorch ML Notebook
                   slug: "pytorch"
                   kubespawner_override:
-                    image: "pangeo/pytorch-notebook:f84546a"
+                    image: "pangeo/pytorch-notebook:2022.10.13"
           kubespawner_override:
             mem_limit: 30G
             mem_guarantee: 24G

diff --git a/config/clusters/m2lines/common.values.yaml b/config/clusters/m2lines/common.values.yaml
@@ -117,15 +117,15 @@ basehub:
               choices:
                 tensorflow:
                   display_name: Pangeo Tensorflow ML Notebook
-                  default: true
                   slug: "tensorflow"
                   kubespawner_override:
-                    image: "pangeo/ml-notebook:f84546a"
+                    image: "pangeo/ml-notebook:2022.10.13"
                 pytorch:
                   display_name: Pangeo PyTorch ML Notebook
+                  default: true
                   slug: "pytorch"
                   kubespawner_override:
-                    image: "pangeo/pytorch-notebook:f84546a"
+                    image: "pangeo/pytorch-notebook:2022.10.13"
           kubespawner_override:
             mem_limit: 30G
             mem_guarantee: 24G