Commit b1054104 authored by biluo.shen's avatar biluo.shen

Use forkserver to avoid NCCL timeout

parent a932ce02
......@@ -36,6 +36,7 @@ def setup(backend, rank, world_size, port):
def mp_start(run):
mp.set_start_method('forkserver')
world_size = int(os.getenv("WORLD_SIZE", "1"))
if world_size == 1:
run(local_rank=0, world_size=world_size)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment