srun jupyter notebook cannot run mpi_relaunch

[TK-21st]

running mpi_relaunch in a jupyter notebook opened with srun does not work.

To work around it:

1. open jupyter notebook without using srun
2. run mpi_relaunch with subprocess inside jupyter cell as follow: import subprocess subprocess.call(['srun','--gres=gpu:1','--pty','python','you_code_that_calls_mpi_relaunch.py'])