diff --git a/modules.sh b/modules.sh
index 76244823bd6a96976be66439626883d9275bdd99..7214be7023876806fdbf2671a72e7f4fb8473236 100644
--- a/modules.sh
+++ b/modules.sh
@@ -1,2 +1,2 @@
 module purge
-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
+module load Stages/2022 GCC/11.2.0 OpenMPI/4.1.2 TensorFlow/2.6.0-CUDA-11.5 Horovod/0.24.3 mpi4py/3.1.3 PyTorch/1.11-CUDA-11.5 scikit-learn/1.0.1