Passer au contenu principal

Deep Learning

Keras

Log into a GPU node:

Sinteractive -p interactive -m 4G -G 1

Check that the GPU is visible:

nvidia-smi

Load parallel modules and python:

module purge
module load gcc cuda cudnn python/3.8.8

Create a virtual environment:

virtualenv -p python venv_keras

TensorFlow

PyTorch