machine=$(cat /etc/FZJ/systemname)
if [[ "$machine" == jurecadc ]]; then
    export CUDA_VISIBLE_DEVICES=0,1,2,3
    #ml use $OTHERSTAGES
    #ml Stages/2020
    #ml GCC/9.3.0
    #ml OpenMPI/4.1.0rc1
    #ml CUDA/11.0
    #ml cuDNN/8.0.2.39-CUDA-11.0
    #ml NCCL/2.8.3-1-CUDA-11.0
    #ml PyTorch
    #ml Horovod/0.20.3-Python-3.8.5
    #ml scikit
    #source /p/project/covidnetx/environments/jureca/bin/activate 
    ml purge
    ml use $OTHERSTAGES
    ml Stages/2020
    ml GCC/10.3.0
    ml OpenMPI/4.1.1
    ml Horovod/0.23.0-Python-3.8.5
    ml scikit
    source /p/project/covidnetx/environments/jureca/bin/activate 
fi
if [[ "$machine" == juwelsbooster ]]; then
    export CUDA_VISIBLE_DEVICES=0,1,2,3
    #ml use $OTHERSTAGES
    #ml Stages/2020
    #ml GCC/9.3.0
    #ml OpenMPI/4.1.0rc1
    #ml CUDA/11.0
    #ml cuDNN/8.0.2.39-CUDA-11.0
    #ml NCCL/2.8.3-1-CUDA-11.0
    #ml PyTorch
    #ml Horovod/0.20.3-Python-3.8.5
    #ml scikit

    #ml Stages/2021
    #ml GCC
    #ml OpenMPI
    #ml CUDA
    #ml cuDNN
    #ml NCCL
    #ml PyTorch
    #ml Horovod
    #ml scikit

    ml purge
    ml use $OTHERSTAGES
    ml Stages/2020
    ml GCC/10.3.0
    ml OpenMPI/4.1.1
    ml Horovod/0.23.0-Python-3.8.5
    ml scikit
    source /p/project/covidnetx/environments/juwels_booster/bin/activate 
fi
if [[ "$machine" == jusuf ]]; then
    ml purge
    ml use $OTHERSTAGES
    ml Stages/2020
    ml GCC/9.3.0
    ml OpenMPI/4.1.0rc1
    ml CUDA/11.0
    ml cuDNN/8.0.2.39-CUDA-11.0
    ml NCCL/2.8.3-1-CUDA-11.0
    ml PyTorch
    ml Horovod/0.20.3-Python-3.8.5
    #ml scikit
    source /p/project/covidnetx/environments/jusuf/bin/activate
fi