Merge pull request #31 from laraPPr/update_tensorflow

Make TensorFlow demo compatible with all the current EESSI repos and versions
EESSI · Feb 8, 2024 · 3789be0 · 3789be0
2 parents 512b688 + f4bb261
commit 3789be0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/TensorFlow/run.sh b/TensorFlow/run.sh
@@ -1,5 +1,10 @@
 #!/bin/bash
+set -e
+
+if [[ $EESSI_CVMFS_REPO == "/cvmfs/software.eessi.io" ]] && [[ $EESSI_VERSION == "2023.06" ]]; then module load TensorFlow/2.13.0-foss-2023a
+elif [[ $EESSI_CVMFS_REPO == "/cvmfs/pilot.eessi-hpc.org" ]] && [[ $EESSI_PILOT_VERSION == "2021.12" ]]; then module load TensorFlow/2.3.1-foss-2020a-Python-3.8.2
+else echo "Don't know which TensorFlow module to load for ${EESSI_CVMFS_REPO}/versions/${EESSI_VERSION}$EESSI_PILOT_VERSION" >&2; exit 1
+fi
 
-module load TensorFlow/2.3.1-foss-2020a-Python-3.8.2
 
 time python TensorFlow-2.x_mnist-test.py