diff --git a/dmlcloud/util/distributed.py b/dmlcloud/util/distributed.py index 4a0423f..13768fe 100644 --- a/dmlcloud/util/distributed.py +++ b/dmlcloud/util/distributed.py @@ -99,7 +99,7 @@ def init_process_group_dummy(): only run on a single gpu. """ store = dist.HashStore() - dist.init_process_group(store=store, rank=0, world_size=1) + dist.init_process_group(store=store, rank=0, world_size=1, backend='gloo') def init_process_group_MPI(ip_idx=0, port=None, **kwargs):