Spaces:
Runtime error
Runtime error
File size: 1,651 Bytes
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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
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
|