Update Dockerfile.cpu

nasa-nccs-hpda · Oct 5, 2024 · 5cc95c2 · 5cc95c2
1 parent d5fdfed
commit 5cc95c2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/requirements/Dockerfile.cpu b/requirements/Dockerfile.cpu
@@ -24,6 +24,7 @@ ENV PYTHONPATH="$PYTHONPATH:/usr/local/ilab"
 ENV PYTHONPATH="$PYTHONPATH:/usr/local/ilab/vhr-cloudmask:/usr/local/ilab/evhr:/usr/local/ilab/core:/usr/local/ilab/srlite/srlite:/usr/local/ilab/srlite:/usr/local/ilab/pyCCDC"
 ENV PYTHONPATH="$PYTHONPATH:/opt/DgStereo/pygeotools:/opt/DgStereo/dgtools"
 ENV PATH="$PATH:/opt/DgStereo/dgtools:/opt/DgStereo/pygeotools:/opt/DgStereo/dgtools/dgtools:/opt/StereoPipeline/bin:/opt/DgStereo/evhr"
+ENV TF_USE_LEGACY_KERAS=1
 
 RUN PROJECT_PATH="/usr/local/ilab" && \
     REDIS_FILE="/etc/profile.d/redis_server.sh" && \
@@ -130,7 +131,8 @@ RUN pip install \
     cubo[ee] \
     git+https://github.com/dshean/pygeotools.git@master \
     git+https://github.com/nasa-nccs-hpda/tensorflow-caney.git@main \
-    tensorflow==2.16.1
+    tensorflow \
+    tf_keras
     #tensorflow==2.14.1
 
 RUN PROJECT_PATH="/usr/local/ilab" && \