-
Notifications
You must be signed in to change notification settings - Fork 0
/
run_GPU.sh
executable file
·45 lines (38 loc) · 1.2 KB
/
run_GPU.sh
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
#! /bin/bash
###### Part 1 ######
#SBATCH --partition=gpu
#SBATCH --account=higgsgpu
#SBATCH --job-name=my_pfn
#SBATCH --qos=debug
#SBATCH --ntasks=2
#SBATCH --mem-per-cpu=4096
#SBATCH --gres=gpu:v100:1
#SBATCH --output=/hpcfs/cepc/higgsgpu/wuzuofei/My_Net/ParticleFlowNetwork/output/my_log/PFN_gputest.log
###### Part 2 ######
ulimit -d unlimited
ulimit -f unlimited
ulimit -l unlimited
ulimit -n unlimited
ulimit -s unlimited
ulimit -t unlimited
srun -l hostname
/usr/bin/nvidia-smi -L
echo "Allocate GPU cards : ${CUDA_VISIBLE_DEVICES}"
nvidia-smi
# >>> conda initialize >>>
# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/hpcfs/cepc/higgsgpu/wuzuofei/miniconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
eval "$__conda_setup"
else
if [ -f "/hpcfs/cepc/higgsgpu/wuzuofei/miniconda3/etc/profile.d/conda.sh" ]; then
. "/hpcfs/cepc/higgsgpu/wuzuofei/miniconda3/etc/profile.d/conda.sh"
else
export PATH="/hpcfs/cepc/higgsgpu/wuzuofei/miniconda3/bin:$PATH"
fi
fi
unset __conda_setup
# <<< conda initialize <<<
conda activate weaver
which python
python -u script_GPUonly/my_train_DDP.py gputest