Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Consider adding support for PyTorch on Apple M1/M2 MPS enabled devices #6043

Closed
dazzag24 opened this issue Feb 22, 2023 · 1 comment
Closed

Comments

@dazzag24
Copy link

dazzag24 commented 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.

@dazzag24 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
@wyli wyli added the Upstream fix required Upstream fix required label Feb 22, 2023
@wyli
Copy link
Contributor

wyli commented Feb 22, 2023

looks like github ci runners don't currently support setup-python with these devices https://github.blog/changelog/2022-08-09-github-actions-self-hosted-runners-now-support-apple-m1-hardware/

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

4 participants