example_submission_script.sh

#!/bin/bash
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --time=00:01:00
#SBATCH --gres=gpu
#SBATCH --partition=develbooster
#SBATCH --account=atmlaml

echo "-------- Outside of the container -----------"
module purge
module load Stages/2020  GCCcore/.10.3.0 PyTorch/1.8.1-Python-3.8.5
python src/hello_world.py

######## How to run with container #########
#echo "-------- Inside of the container ------------"
#module purge
#module load Singularity-Tools
#singularity run pytorch22.02.sif python3 src/hello_world.py
#echo "--------------------------------------------"