-
Notifications
You must be signed in to change notification settings - Fork 54
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
Support Apple's MPS backend #1053
Comments
First few tests (PyTorch 1.13):
|
This issue is stale because it has been open for 60 days with no activity. |
Branch 1053-Support_Apple_s_MPS_backend created! |
This issue is stale because it has been open for 60 days with no activity. |
This issue was closed because it has been inactive for 60 days since being marked as stale. |
In progress #1129 |
This issue is stale because it has been open for 60 days with no activity. |
Since version 1.12, PyTorch supports GPU acceleration via Apple's Metal Performance Shaders (MPS) backend.
https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/
Feature functionality
device="gpu"
should be a valid argument forht.factories
if the MPS backend is available for torch versions >= 1.12.Ongoing research on branch
features/1053-support-Apple-silicon-GPUs
, will document here.The text was updated successfully, but these errors were encountered: