Spaces:
Runtime error
Runtime error
File size: 1,319 Bytes
c81908d 23d6920 c81908d be61cf2 c81908d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
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
|