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
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