From 46345fee4838a13a11ec4d6215a604608528122d Mon Sep 17 00:00:00 2001
From: Carl <cboettig@gmail.com>
Date: Tue, 16 Jan 2024 12:50:50 -0800
Subject: [PATCH] wip

---
 images/Dockerfile             | 2 +-
 images/Dockerfile.gpu         | 1 +
 jupyterhub/public-config.yaml | 4 ++--
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/images/Dockerfile b/images/Dockerfile
index 0d0858e..b123886 100644
--- a/images/Dockerfile
+++ b/images/Dockerfile
@@ -17,9 +17,9 @@ RUN chown -R root:staff ${VIRTUAL_ENV} && chmod -R ug+rw ${VIRTUAL_ENV}
 USER ${NB_USER}
 RUN usermod -s /bin/bash ${NB_USER}
 RUN python3 -m pip install -r /tmp/requirements.txt
+RUN python3 -m pip install jupyter-tensorboard-proxy
 RUN python3 -m ipykernel install --user --name=spatial
 
-
 WORKDIR /home/${NB_USER}
 ENV PATH $VIRTUAL_ENV/bin:$PATH
 
diff --git a/images/Dockerfile.gpu b/images/Dockerfile.gpu
index 9b4956f..2e95a80 100644
--- a/images/Dockerfile.gpu
+++ b/images/Dockerfile.gpu
@@ -17,6 +17,7 @@ RUN chown -R root:staff ${VIRTUAL_ENV} && chmod -R ug+rw ${VIRTUAL_ENV}
 USER ${NB_USER}
 RUN usermod -s /bin/bash ${NB_USER}
 RUN python3 -m pip install -r /tmp/requirements.txt
+RUN python3 -m pip install jupyter-tensorboard-proxy
 RUN python3 -m ipykernel install --user --name=spatial
 
 
diff --git a/jupyterhub/public-config.yaml b/jupyterhub/public-config.yaml
index a714c08..bf62007 100644
--- a/jupyterhub/public-config.yaml
+++ b/jupyterhub/public-config.yaml
@@ -63,12 +63,12 @@ singleuser:
                 mem_guarantee: 31861460992
                 mem_limit: 31861460992
       kubespawner_override:
-        image: ghcr.io/boettiger-lab/k8s-gpu:latest
+        image: ghcr.io/boettiger-lab/k8s:latest
         default_url: /lab # change to /rstudio to go straight to RStudio
         working_dir: /home/rstudio
     - display_name: "GPU"
       slug: gpu 
-      description: Jupyter + R environment with many geospatial libraries pre-installed
+      description: Jupyter + R with GPU 
       profile_options:
         resource_allocation:
           display_name: Resource Allocation