Deep Learning

Keras

Log into a GPU node:

Sinteractive -p interactive -m 4G -G 1

Check that the GPU is visible:
nvidia-smi
module purge
module load gcc cuda cudnn python/3.8.8

TensorFlow

PyTorch