#!/bin/bash
#SBATCH -p gpu                           # Specify partition [Compute/Memory/GPU]
#SBATCH -N 1 -c 16   			         # Specify number of nodes and processors per task
#SBATCH --gpus-per-node=4		         # Specify number of GPU per task
#SBATCH --ntasks-per-node=4		         # Specify tasks per node
#SBATCH -t 120:00:00                     # Specify maximum time limit (hour: minute: second)
#SBATCH -A ltxxxxxx               	     # Specify project name
#SBATCH -J JOBNAME               	     # Specify job name
module load Apptainer/1.1.6              # Load the Apptainer module
apptainer exec --nv -B $PWD:$PWD tensorflow_2.317.10-gpu.sif python3 MNIST.py       # Run your program

username@lanta:~> ml Apptainer
username@lanta:~> apptainer exec -B $PWD:$PWD tensorflow_2.317.10-gpu.sif python Setup.py