Fixes a bug in dockerfile

keisen · May 20, 2021 · 08a9415 · 08a9415
1 parent 03844f7
commit 08a9415
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 13 deletions.
diff --git a/dockerfiles/cpu.Dockerfile b/dockerfiles/cpu.Dockerfile
@@ -17,10 +17,7 @@ RUN export JUPYTER_HOME=/etc/jupyter                               && \
                 {'shell_command': ['/bin/bash']}" >> $JUPYTER_CONF
 
 # Install essential python libraries
-RUN pip install --no-cache-dir --upgrade pip                && \
-    pip install --no-cache-dir --upgrade setuptools         && \
-    pip install --no-cache-dir                              && \
-      tf-keras-vis[develop,examples]==$TF_KERAS_VIS_VERSION && \
-      jupyterlab==2.*
+RUN pip install --no-cache-dir \
+      tf-keras-vis[develop,examples]==$TF_KERAS_VIS_VERSION
 
-CMD [ "jupyter", "lab" ]
+CMD jupyter lab
diff --git a/dockerfiles/gpu.Dockerfile b/dockerfiles/gpu.Dockerfile
@@ -1,6 +1,5 @@
 FROM tensorflow/tensorflow:2.4.1-gpu
 
-# Default ENV Settings
 ARG TF_KERAS_VIS_VERSION=0.6.0
 ARG JUPYTER_ALLOW_IP="0.0.0.0"
 ARG JUPYTER_TOKEN=""
@@ -17,10 +16,7 @@ RUN export JUPYTER_HOME=/etc/jupyter                               && \
                 {'shell_command': ['/bin/bash']}" >> $JUPYTER_CONF
 
 # Install essential python libraries
-RUN pip install --no-cache-dir --upgrade pip                && \
-    pip install --no-cache-dir --upgrade setuptools         && \
-    pip install --no-cache-dir                              && \
-      tf-keras-vis[develop,examples]==$TF_KERAS_VIS_VERSION && \
-      jupyterlab==2.*
+RUN pip install --no-cache-dir \
+      tf-keras-vis[develop,examples]==$TF_KERAS_VIS_VERSION
 
-CMD [ "jupyter", "lab"]
+CMD jupyter lab