test GPU and newer JupyterHub

det-lab · Mar 17, 2024 · f7d2e9f · f7d2e9f
1 parent b54c17c
commit f7d2e9f
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 20 deletions.
diff --git a/config_standard_storage.yaml b/config_standard_storage.yaml
@@ -32,12 +32,12 @@ hub:
 
 singleuser:
   memory:
-      guarantee: 15G
-      limit: 15G
+      guarantee: 1G
+      limit: 1G
   cpu:
-      guarantee: 6
+      guarantee: 1
   profileList:
-    - display_name: "Default: 6 CPUs and 15 GB RAM"
+    - display_name: "Default: 1 CPUs and 1 GB RAM"
       default: True
     - display_name: "Tiny: 1 CPUs and 5 GB RAM"
       kubespawner_override:
@@ -54,26 +54,39 @@ singleuser:
         cpu_guarantee: 24
         mem_limit: 60G
         mem_guarantee: 60G
+    - display_name: "GPU Server"
+      description: "Spawns a notebook server with access to a GPU"
+      kubespawner_override:
+        tolerations:
+            - key: 'gpu'
+              operator: 'Equal'
+              value: 'true'
+              effect: 'NoSchedule'
+        extra_resource_limits:
+          nvidia.com/gpu: "1"
   storage:
     type: dynamic
     capacity: 10Gi
     dynamic:
       storageClass: cinder-csi
-    extraVolumes:
-        - name: cvmfs-nfs-volume
-          nfs:
-            # for Magnum
-            # server: 10.254.77.77
-            # for Kubespray
-            server: 10.233.46.63
-            path: /
-    extraVolumeMounts:
-        - name: cvmfs-nfs-volume
-          mountPath: /cvmfs
-          readOnly: false
+        #   extraVolumes:
+        #       - name: cvmfs-nfs-volume
+        #         nfs:
+        #           # for Magnum
+        #           # server: 10.254.77.77
+        #           # for Kubespray
+        #           server: 10.233.46.63
+        #           path: /
+        #   extraVolumeMounts:
+        #       - name: cvmfs-nfs-volume
+        #         mountPath: /cvmfs
+        #         readOnly: false
+        # image:
+        #   name: "zonca/docker-jupyter-cdms-light"
+        #   tag: "22.06.3"
   image:
-    name: "zonca/docker-jupyter-cdms-light"
-    tag: "22.06.3"
+      name: zonca/nvidia-tensorflow-jupyterhub
+      tag: "23.1.5"
   defaultUrl: "/lab"
   # default is 300s, sometimes Jetstream volumes are slow to attach
   startTimeout: 600

diff --git a/install_jhub.sh b/install_jhub.sh
@@ -3,5 +3,7 @@ NAMESPACE=jhub
 
 helm upgrade --install $RELEASE jupyterhub/jupyterhub \
       --namespace $NAMESPACE  \
-      --values config_standard_storage.yaml --values cdms_secrets.yaml \
-      --version 3.1.0
+      --create-namespace \
+      --version 3.2.1 \
+      --debug \
+      --values config_standard_storage.yaml --values secrets.yaml