call dist.is_initialized() after checking the availability (#128)

688e6abe · Kai Chen · GitHub · 22ba143d · 688e6abe
Unverified Commit 688e6abe authored Sep 07, 2019 by Kai Chen Committed by GitHub Sep 07, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

mmcv/runner/utils.py mmcv/runner/utils.py +4 -1

No files found.
--- a/mmcv/runner/utils.py
+++ b/mmcv/runner/utils.py
@@ -18,7 +18,10 @@ def get_dist_info():
    if torch.__version__ < '1.0':
        initialized = dist._initialized
    else:
-        initialized = dist.is_initialized()
+        if dist.is_available():
+            initialized = dist.is_initialized()
+        else:
+            initialized = False
    if initialized:
        rank = dist.get_rank()
        world_size = dist.get_world_size()