here we go!

boettiger-lab · Sep 21, 2024 · 21852a0 · 21852a0
1 parent 71a39ff
commit 21852a0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/images/Dockerfile.pytorch b/images/Dockerfile.pytorch
@@ -1,6 +1,8 @@
-FROM quay.io/jupyter/pytorch-notebook:cuda12-2024-04-22
+FROM quay.io/jupyter/pytorch-notebook:cuda12-ubuntu-24.04
 COPY jupyter-ai.yml environment.yml
-RUN conda env update --file environment.yml
+
+RUN conda update -n base -c conda-forge conda && \
+    conda env update --file environment.yml
 
 USER root
 RUN curl -L https://ollama.com/download/ollama-linux-amd64 -o /usr/local/bin/ollama && chmod +x /usr/local/bin/ollama

diff --git a/images/jupyter-ai.yml b/images/jupyter-ai.yml
@@ -6,6 +6,7 @@ dependencies:
   - jupyter-ai
   - jupyter-resource-usage
   - jupyter-vscode-proxy
+  - pip
   - pip:
     - jupyter-tensorboard-proxy