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/20220928
ml-gpu python3