Merge pull request #26 from stfc/fix-missing-packages-tf

Use mamba install to install required packages for tensorflow notebook
stfc · Aug 23, 2023 · 3a83ca7 · 3a83ca7
2 parents e01a186 + c44e8d0
commit 3a83ca7
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/jupyter-tensorflow-notebook/Dockerfile b/jupyter-tensorflow-notebook/Dockerfile
@@ -18,16 +18,12 @@ USER ${NB_UID}
 # Required for Tensorboard on a remote container
 RUN mamba install --yes \
     'jupyter-server-proxy' && \
+    # SciML requirements
+    mamba install --yes \
+    'gym=0.22' 'pygame' && \
     mamba clean --all -f -y && \
     fix-permissions "${CONDA_DIR}" && \
     fix-permissions "/home/${NB_USER}"
 
-# SciML requirements
-RUN pip install --user --no-cache-dir \
-    gym==0.25.2 \
-    pygame && \
-    fix-permissions "${CONDA_DIR}" && \
-    fix-permissions "/home/${NB_USER}"
-
 # Note, the trailing slash is important!
 ENV TENSORBOARD_PROXY_URL=/user-redirect/proxy/6006/