Currently the SLURM builder only supports JSC out-of-the-box. However, it would be nice with some functionality so that you can pick which HPC you are using in e.g. the config. Maybe also adding a file with the necessary setup variables etc.
Necessary modules for JSC:
Stages/2024 GCC OpenMPI CUDA/12 cuDNN MPI-settings/CUDA Python CMake HDF5 PnetCDF libaio mpi4py
Necessary modules for Vega:
Python/3.11.5-GCCcore-13.2.0 CMake/3.24.3-GCCcore-11.3.0 mpi4py OpenMPI CUDA/12.3 GCCcore/11.3.0 NCCL cuDNN/8.9.7.29-CUDA-12.3.0 UCX-CUDA/1.15.0-GCCcore-13.2.0-CUDA-12.3.0
Notice, however, that the version of GCCcore varies slightly in the necessary modules for vega.
Currently the SLURM builder only supports JSC out-of-the-box. However, it would be nice with some functionality so that you can pick which HPC you are using in e.g. the config. Maybe also adding a file with the necessary setup variables etc.
Necessary modules for JSC:
Necessary modules for Vega:
Notice, however, that the version of
GCCcorevaries slightly in the necessary modules for vega.