File size: 596 Bytes
c614b0f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
#!/bin/sh
# Example: ./sub.sh tmp_proj 0,1,2,3 3 --> Use 0,1,2,3 for training, release GPUs, use GPU:3 for inference.
module load gcc/11.2.0 cuda/11.8 cudnn/8.6.0_cu11x && cpu_core_num=6
# module load compilers/cuda/11.8 compilers/gcc/12.2.0 cudnn/8.4.0.27_cuda11.x && cpu_core_num=32
export PYTHONUNBUFFERED=1
method=${1:-"BSL"}
devices=${2:-0}
gpu_num=$(($(echo ${devices%%,} | grep -o "," | wc -l)+1))
sbatch --nodes=1 -p vip_gpu_ailab -A ai4bio \
--gres=gpu:${gpu_num} --ntasks-per-node=1 --cpus-per-task=$((gpu_num*cpu_core_num)) \
./train_test.sh ${method} ${devices}
hostname
|