Fix notebook `misconfiguration` error
# What does this PR do ?

The notebook is complaining about Trainer `misconfiguration` error because it tries to spawn processes in the python interactive session. Fixed it by using torch elastic environment. 