scripts/init.sh

machine=$(cat /etc/FZJ/systemname)
if [[ "$machine" == jurecadc ]]; then
    export CUDA_VISIBLE_DEVICES=0,1,2,3
    ml purge
    ml use $OTHERSTAGES
    ml Stages/2022
    ml GCC/11.2.0
    ml OpenMPI/4.1.2
    ml CUDA/11.5
    ml cuDNN/8.3.1.22-CUDA-11.5
    
    ml NCCL/2.11.4-CUDA-11.5
    #ml NCCL/2.12.7-1-CUDA-11.5
    
    ml PyTorch/1.11-CUDA-11.5
    ml Horovod/0.24
    ml torchvision/0.12.0
    source /p/project/covidnetx/environments/jureca_2022/bin/activate 
fi
if [[ "$machine" == juwelsbooster ]]; then
    export CUDA_VISIBLE_DEVICES=0,1,2,3
    ml purge
    ml use $OTHERSTAGES
    ml Stages/2022
    ml GCC/11.2.0
    ml OpenMPI/4.1.2
    ml CUDA/11.5
    ml cuDNN/8.3.1.22-CUDA-11.5
    ml NCCL/2.12.7-1-CUDA-11.5
    ml PyTorch/1.11-CUDA-11.5
    ml Horovod/0.24
    ml torchvision/0.12.0
    source /p/project/covidnetx/environments/juwels_booster_2022/bin/activate 
fi
if [[ "$machine" == hdfml ]]; then
    export CUDA_VISIBLE_DEVICES=0,1,2,3
    ml purge
    ml use $OTHERSTAGES
    ml Stages/2022
    ml GCC/11.2.0
    ml OpenMPI/4.1.2
    ml CUDA/11.5
    ml cuDNN/8.3.1.22-CUDA-11.5
    ml NCCL/2.12.7-1-CUDA-11.5
    ml PyTorch/1.11-CUDA-11.5
    ml Horovod/0.24
    ml torchvision/0.12.0
    source envs/hdfml/bin/activate
fi
if [[ "$machine" == jusuf ]]; then
    echo not supported
fi