cluster_job.pbs

#!/bin/bash
#$ -N qtest
#$ -j y                         # Merge the IO to standard IO
#$ -pe smp 6
#$ -cwd                         # Run in current directory. Otherwise defaults to home folder
#$ -V
#$ -l h_vmem=6G
#$ -l gpu=2

# For more options, follow the steps from http://bioinformatics.mdc-berlin.de/intro2UnixandSGE/index.html
# qhost -F shows details of all the nodes

echo "Running on: "
echo ${HOSTNAME}
echo ${SGE_O_WORKDIR}
pwd
echo
echo "Program Output begins: "

cd ${SGE_O_WORKDIR}
nvidia-smi

# ============================================================== #
## Alternate Script
#!/bin/sh
# Torque script to submit CUDA C/C++ programs.
# Torque directives
#PBS -N testpy
#PBS -q gpu
#PBS -l walltime=00:05:00,mem=400mb,nodes=1:ppn=1 
#PBS -V

# User Directives
cd htd/
#git checkout master
#CUDA_VISIBLE_DEVICES=0,1 ~/anaconda3/envs/myenv/bin/python3 main.py > hier_check.log
#CUDA_VISIBLE_DEVICES=0,1 ~/anaconda3/envs/myenv/bin/python3 main_acts.py -embed 272 -heads 8 -hid 96 -l_e1 4 -l_e2 2 -l_d 4 -d 0.0868 -bs 16 -e 30 -model HIER++ > hier++_act.log
git checkout joint
CUDA_VISIBLE_DEVICES=0,1 ~/anaconda3/envs/myenv/bin/python3 main.py > joint.log
#End of script