Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2024-03-15 00:43:46 +00:00
parent cd63115a76
commit a25eaf8be8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/lightning/fabric/utilities/system_check.py
View File

@ -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: