run.slurm

#!/bin/bash
#SBATCH --partition=gpuq                    # need to set 'gpuq' or 'contrib-gpuq'  partition
#SBATCH --qos=gpu                           # need to select 'gpu' QOS or other relvant QOS
#SBATCH --job-name=python-gpu
#SBATCH --output="/scratch/%u/Clinical Risk Prediction/outputs/%x-%N-%j.out"   # Output file
#SBATCH --error="/scratch/%u/Clinical Risk Prediction/errors/%x-%N-%j.err"    # Error file
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=4                # number of cores needed
#SBATCH --gres=gpu:A100.80gb:1                # up to 8; only request what you need
#SBATCH --mem-per-cpu=8000M                 # memory per CORE; total memory is 1 TB (1,000,000 MB)
#SBATCH --export=ALL 
#SBATCH --time=0-01:30:00                   # D-HH:MM:SS; please choose carefully

set -x
umask 0027

# change directory to the root of the project
cd "/scratch/apathak2/Clinical Risk Prediction"

# Add the current directory (project root) to the PYTHONPATH
export PYTHONPATH=$(pwd):$PYTHONPATH

# to see ID and state of GPUs assigned
nvidia-smi

module load gnu10                           
module load python

python src/main.py