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
PyTorch now runs on Apple M1 MPS devices. This could open up access to MONAI to a wider range of users without the need for a dedicated GPU device.
However the current support for MPS is not 100% complete so this is very likely not feasible right now.
Others have already run into operations that are not supported e.g Conv3D() and the PyTorch team have stated that they are planning on adding support.
See this which tracks adding support for the missing operations for the MPS backend.
The text was updated successfully, but these errors were encountered:
dazzag24
changed the title
Consider adding support for Pytorch on Apple M1 MPS devices
Consider adding support for PyTorch on Apple M1/M2 MPS enabled devices
Feb 22, 2023
PyTorch now runs on Apple M1 MPS devices. This could open up access to MONAI to a wider range of users without the need for a dedicated GPU device.
However the current support for MPS is not 100% complete so this is very likely not feasible right now.
Others have already run into operations that are not supported e.g Conv3D() and the PyTorch team have stated that they are planning on adding support.
See this which tracks adding support for the missing operations for the MPS backend.
The text was updated successfully, but these errors were encountered: