...

*** Make sure that Prof. Keller has added you to both the spin and spinquest groups in Rivanna. ***

Using GPU's on Rivanna

#!/bin/bash
#SBATCH -A <allocation_name>
#SBATCH -p gpu
#SBATCH --gres=gpu:a100
#SBATCH --constraint=a100_80gb
#SBATCH -c 4
#SBATCH -t 1:00:00
#SBATCH -J test
#SBATCH -o test.out
#SBATCH -e test.err
#SBATCH --mem=256000

module purge
module load apptainer tensorflow/2.13.0

apptainer run --nv $CONTAINERDIR/tensorflow-2.13.0.sif <file_name>.py

Using Tensorflow with BKM2002-Formulation

...

2. Run the following commands on your terminal
                    module load anaconda/2020.11-py3.8
                    module load singularity/3.7.1
                    module load tensorflow/2.1.0-py37
                     following step is needed to run only once (it will copy the relevant .sif file to your /home directory)
                    cp  $CONTAINERDIR$$CONTAINERDIR/tensorflow-2.1.0-py37.sif /home/$USER  

...

2. Run the following commands on your terminal
                    module load anaconda/2020.11-py3.8
                    module load singularity/3.7.1
                    module load pytorch/1.8.1
                     following step is needed to run only once (it will copy the relevant .sif file to your /home directory)
                    cp  CONTAINERDIR$$CONTAINERDIR/pytorch-1.8.1.sif /home/$USER

...