From a25eaf8be8e58559ad2f486167832905497b76ca Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 15 Mar 2024 00:43:46 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/lightning/fabric/utilities/system_check.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/lightning/fabric/utilities/system_check.py b/src/lightning/fabric/utilities/system_check.py index ed78983022..5453ecae13 100644 --- a/src/lightning/fabric/utilities/system_check.py +++ b/src/lightning/fabric/utilities/system_check.py @@ -23,8 +23,8 @@ from pathlib import Path import torch import torch.distributed import torch.multiprocessing as mp -from torch.multiprocessing.spawn import ProcessRaisedException from lightning_utilities.core.imports import RequirementCache +from torch.multiprocessing.spawn import ProcessRaisedException _psutil_available = RequirementCache("psutil") _logger = logging.getLogger(__name__) @@ -55,7 +55,7 @@ def main(timeout: int = 60) -> None: "NCCL_NET_PLUGIN": "none", } _print0( - f"The multi-GPU NCCL test did not succeed." + "The multi-GPU NCCL test did not succeed." " It looks like there is an issue with your multi-GPU setup." " Now trying to run again with NCCL features disabled." ) @@ -94,7 +94,7 @@ def _check_cuda_distributed(timeout: int) -> bool: _logger.debug(str(e)) success = False break - + time.sleep(1) if not success: