sbatch check gpu

testScript_gpu_check.sh

#!/bin/bash
#SBATCH -t 20                               # time limit of job, in minutes.. just using now for testing
#SBATCH -p gpu                              # which queue to submit to
#SBATCH -n 40                                   # TEST FROM DR M
#SBATCH -N 2                                    # TEST FROM DR M
#SBATCH -o gpu_python_%j.out                    # File to which STDOUT will be written
#SBATCH -e gpu_python_%j.err                    # File to which STDERR will be written
#SBATCH --mail-type=END                     # Type of email notification- BEGIN,END,FAIL,ALL
#SBATCH [email protected]    # Email to which notifications will be sent

THEANO_FLAGS=mode=FAST_RUN,device=gpu,floatX=float32 python test_GPU_or_CPU.py