Updated 2019-10-18

pytorch.pbs

#PBS -N pytorchTest
#PBS -l nodes=1:ppn=2:gpus=1
#PBS -l pmem=2gb
#PBS -l walltime=10:00
#PBS -q testflight-gpu
#PBS -j oe
#PBS -o pytorchTest.out

cd $PBS_O_WORKDIR
module load pytorch/1.2
source activate $PYTORCHROOT

cp -r $PYTORCHROOT/examples/word_language_model .
cd word_language_model
python main.py --cuda --epochs 6