Add import os to example code block (#3378)

Co-authored-by: sekyondaMeta <127536312+sekyondaMeta@users.noreply.github.com>

Author: FORTFANOP

intermediate_source/ddp_tutorial.rst

@@ -311,6 +311,7 @@ Let's still use the Toymodel example and create a file named ``elastic_ddp.py``.
 
 .. code:: python
 
+    import os
     import torch
     import torch.distributed as dist
     import torch.nn as nn