Para lanzar Jupyter es necesario crear un archivo con el siguiente script de SLURM:

jupyter.sbatch

#!/bin/bash
#SBATCH --job-name=jupyter-notebook
#SBATCH --nodes=1
#SBATCH --time=4:00
#SBATCH --partition=tara-2N-4H
#SBATCH --output=notebook.txt
#SBATCH --gres=gpu:kepler:2
 
module add python3/3.8.3 toolkits/python3-base/3.8.3
module add toolkits/tensorflow/2.2.0
module add image-processing/opencv/4.3.0
module add cuda/10.1.243
 
hostname
 
jupyter notebook --ip 0.0.0.0 --port 8888

• Primero se ejecuta el script de SLURM:

sbatch jupyter.sbatch

• Posterior a ejecutar el script de SLURM, es necesario establecer en que nodo se está ejecutando Jupyer, para esto ejecute:

scontrol show jobid -dd <job id> | grep -e "\bNodeList="

• Finalmente es necesario redirigir el puerto mediante SSH:

ssh -N -L :8080:localhost:8888 <nodo de slurm>