From 5fb3e51a3897e2a60aef71b0d6119ed0544c6112 Mon Sep 17 00:00:00 2001
From: xuetaowave <cxt@cxt-win>
Date: Sun, 18 Feb 2024 18:37:24 +0800
Subject: [PATCH] update

---
 multi_gpu.py | 7 +++++++
 1 file changed, 7 insertions(+)
 create mode 100644 multi_gpu.py

diff --git a/multi_gpu.py b/multi_gpu.py
new file mode 100644
index 0000000..7ad54a7
--- /dev/null
+++ b/multi_gpu.py
@@ -0,0 +1,7 @@
+
+from utils import logging_utils
+logging_utils.config_logger()
+
+import torch.distributed as dist
+dist.init_process_group(backend='nccl',
+                            init_method='env://')
\ No newline at end of file
-- 
GitLab