-
Notifications
You must be signed in to change notification settings - Fork 30
/
pbsbatch.sh
46 lines (41 loc) · 1.82 KB
/
pbsbatch.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
46
#!/bin/sh
### Set the job name (for your reference)
#PBS -N extractfromzip
### Set the project name, your department code by default
#PBS -P cse
### Request email when job begins and ends, don't change anything on the below line
#PBS -m bea
### Specify email address to use for notification, don't change anything on the below line
#PBS -M [email protected]
#### Request your resources, just change the numbers
#PBS -l select=3:ncpus=12:ngpus=1:mem=16G
### Specify "wallclock time" required for this job, hhh:mm:ss
#PBS -l walltime=24:00:00
#PBS -l software=PYTHON
# After job starts, must goto working directory.
# $PBS_O_WORKDIR is the directory from where the job is fired.
echo "==============================="
echo $PBS_JOBID
cat $PBS_NODEFILE
echo "==============================="
cd $PBS_O_WORKDIR
echo $PBS_O_WORKDIR
module () {
eval `/usr/share/Modules/$MODULE_VERSION/bin/modulecmd bash $*`
}
module load apps/anaconda/3
source activate ~/myenv
module unload apps/anaconda/3
module load compiler/python/3.6.0/ucs4/gnu/447
module load pythonpackages/3.6.0/ucs4/gnu/447/pip/9.0.1/gnu
module load pythonpackages/3.6.0/ucs4/gnu/447/setuptools/34.3.2/gnu
module load pythonpackages/3.6.0/ucs4/gnu/447/wheel/0.30.0a0/gnu
module load pythonpackages/3.6.0/numpy/1.16.1/gnu
module load pythonpackages/3.6.0/pandas/0.23.4/gnu
module load compiler/cuda/9.2/compilervars
module load compiler/gcc/9.1.0
module load apps/pythonpackages/3.6.0/tensorflow/1.9.0/gpu
module load pythonpackages/3.6.0/tensorflow_tensorboard/1.10.0/gnu
module load apps/pythonpackages/3.6.0/keras/2.2.2/gpu
module load pythonpackages/3.6.0/tqdm/4.25.0/gnu
python3 codefiles2/multi-gpu-train.py --data_dir=codefiles2/data --saved_models_dir=saved_models --log_dir=codefiles2/training_logs --rnn_size=4096 --batch_size=128 --seq_length=256 --embedding_size=64 --num_gpus=2