Interactive Session with a GPU

To get an interactive session for an hour on one node with 1 CPU and 1 GPU:

srun --time=01:00:00 --nodes=1 --cpus-per-task=1 --mem=8G --partition=gpu-interactive --gres=gpu:1 --pty /usr/bin/bash

Any of the arguments you can use with sbatch can be provided here too. You can also use the job script generator.

When you do this you should consider protecting your session with tmux or screen.

Then run the following to start python within the ml-gpu container:

module load ml-gpu/20230427
ml-gpu python3