diff --git a/modules.sh b/modules.sh
index 6512ef50610da295a239302ce077022223222cb1..76244823bd6a96976be66439626883d9275bdd99 100644
--- a/modules.sh
+++ b/modules.sh
@@ -1,2 +1,2 @@
 module purge
-module load GCC/10.3.0  ParaStationMPI/5.4.10-1  Python/3.8.5
+module load Stages/2020  GCC/10.3.0  OpenMPI/4.1.1 TensorFlow/2.5.0-Python-3.8.5 Horovod/0.23.0-Python-3.8.5  mpi4py/3.0.3-Python-3.8.5 PyTorch/1.8.1-Python-3.8.5 scikit/2021-Python-3.8.5