You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
MPSAccelerator can not run on your system since the accelerator is not available. The following accelerator(s) is available and can be passed into accelerator argument of Trainer: ['cpu'].
🐛 Bug
Raised from
accelerator_connector.py::AcceleratorConnector#_set_parallel_devices_and_init_accelerator
:https://github.com/Lightning-AI/lightning/blob/2415834aaf4f215ddb0d618bf415405d670f9114/src/pytorch_lightning/trainer/connectors/accelerator_connector.py
Availability determined here:
https://github.com/Lightning-AI/lightning/blob/2415834aaf4f215ddb0d618bf415405d670f9114/src/pytorch_lightning/accelerators/mps.py
Check is wrong due to platform check here expecting
arm
(my M1 Max reports platformarm64
):https://github.com/Lightning-AI/lightning/blob/master/src/pytorch_lightning/accelerators/mps.py#L27
To Reproduce
Expected behavior
Program should complete without raising an error.
Environment
cc @tchaton @rohitgr7 @akihironitta @justusschock
The text was updated successfully, but these errors were encountered: