slurm script to run mnist_example

mnist.cmd

#!/bin/bash
#SBATCH --gres=gpu:1 
#one GPU, asking for only one CPU core since we're offloading to GPU
#SBATCH -t 00:05:00
#SBATCH --mail-type=begin
#SBATCH --mail-type=end
#SBATCH [email protected]

module load anaconda3
# assumes you have a conda environment named tf with tensorflow and matplotlib, i.e.
# conda create -n tf tensorflow matplotlib
source activate tf
python mnist_classify.py