|
@@ -238,7 +238,7 @@ if __name__ == "__main__":
|
|
|
backend = 'gloo' if os.name == 'nt' else 'nccl'
|
|
|
|
|
|
# 使用文件初始化方法
|
|
|
- init_method = 'file:///tmp/shared_file' # 文件路径需要所有进程都能访问
|
|
|
+ init_method = f'../tmp' # env:// # 文件路径需要所有进程都能访问
|
|
|
dist.init_process_group(backend=backend, init_method=init_method)
|
|
|
|
|
|
print(f"Initialized distributed training with backend: {backend}")
|