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"