#!/bin/bash
#SBATCH -Jpytorch_tensors                # job name
#SBATCH -A [Account]                     # account to which job is charged, ex: GT-gburdell3
#SBATCH -N 1 --ntasks-per-node=1         # number of nodes and cores per node required
#SBATCH --mem-per-cpu=2G                 # memory per core
#SBATCH -t10                             # duration of the job (ex: 10 min)
#SBATCH -oReport-%j.out                  # output file name

cd $SBATCH_O_WORKDIR

module load anaconda3/2022.05.01
conda activate my_pytorch

python pytorch_tensors.py