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

Add GPU Support for Apple Silicon Macs #1481

Closed
wants to merge 3 commits into from
Closed

Add GPU Support for Apple Silicon Macs #1481

wants to merge 3 commits into from

Conversation

NripeshN
Copy link

No description provided.

@jongwook
Copy link
Collaborator

Hi, this is duplicate of #382, where we have various reports that it either fails to run some operations or performance that's not very promising. Looks like it's doing better on pytorch 2.x; I'll plan to merge #382 after confirming that MPS is works reliably and fast enough.

@NripeshN
Copy link
Author

NripeshN commented Jun 30, 2023

Hi @jongwook, this pull request may pass the failing operations. PyTorch should be installed with MPS compatibility for torch.backends.mps.is_available() to work. Otherwise, it would throw out an error. Wrapping it around try and catch statement would fix this issue.

@NripeshN NripeshN closed this Jun 30, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
2 participants