diff --git a/modules.sh b/modules.sh index 76244823bd6a96976be66439626883d9275bdd99..cb649a0e9af9353cd8357c4691151b7ed6536149 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 OpenMPI TensorFlow Horovod mpi4py PyTorch scikit-learn