From 2b50cbb4d3761c5fe03f137deb86fb2b2a6f2aa8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Fri, 9 Sep 2022 21:08:34 +0200 Subject: [PATCH] Avoid instantiating every accelerator in the registry (#14591) * Avoid instantiating every accelerator in the registry when listing available ones --- .../trainer/connectors/accelerator_connector.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/pytorch_lightning/trainer/connectors/accelerator_connector.py b/src/pytorch_lightning/trainer/connectors/accelerator_connector.py index f3be6caa5b..d50fc86140 100644 --- a/src/pytorch_lightning/trainer/connectors/accelerator_connector.py +++ b/src/pytorch_lightning/trainer/connectors/accelerator_connector.py @@ -524,7 +524,9 @@ class AcceleratorConnector: if not self.accelerator.is_available(): available_accelerator = [ - acc_str for acc_str in self._accelerator_types if AcceleratorRegistry.get(acc_str).is_available() + acc_str + for acc_str in self._accelerator_types + if AcceleratorRegistry[acc_str]["accelerator"].is_available() ] raise MisconfigurationException( f"{self.accelerator.__class__.__qualname__} can not run on your system"