Update startscript.sh

interTwin-eu · Jun 7, 2024 · f656fa3 · f656fa3
1 parent 17c8e94
commit f656fa3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/use-cases/mnist/tensorflow/startscript.sh b/use-cases/mnist/tensorflow/startscript.sh
@@ -28,6 +28,11 @@ ml Stages/2024 GCC/12.3.0 OpenMPI CUDA/12 MPI-settings/CUDA Python/3.11 HDF5 Pne
 # shellcheck source=/dev/null
 source ~/.bashrc
 
+# Using legacy (2.16) version of Keras
+# Latest version with TF (2.16) installs Keras 3.3
+# which returns an error for multi-node execution
+export TF_USE_LEGACY_KERAS=1
+
 # ON LOGIN NODE download datasets:
 # ../../../.venv-tf/bin/itwinai exec-pipeline --config pipeline.yaml --pipe-key pipeline --steps 0
 source ../../../envAItf_hdfml/bin/activate