Fixes issue #7774 (#8322)

pytorch · Oct 30, 2024 · 8177447 · 8177447
1 parent c9e02e9
commit 8177447
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/docs/source/perf/ddp.md b/docs/source/perf/ddp.md
@@ -77,8 +77,7 @@ import torch_xla.runtime as xr
 import torch_xla.distributed.xla_backend
 
 def setup(rank, world_size):
-    os.environ['MASTER_ADDR'] = 'localhost'
-    os.environ['MASTER_PORT'] = '12355'
+    os.environ['PJRT_DEVICE'] = 'TPU'
 
     # initialize the xla process group
     dist.init_process_group("xla", rank=rank, world_size=world_size)