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

pytorch: enable gpu for apple mps #2453

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

pytorch: enable gpu for apple mps #2453

wants to merge 1 commit into from

Conversation

mqy
Copy link

@mqy mqy commented Mar 9, 2023

This commit roughly enables mps for the d2l's pytorch implementation.
Tested on mac 13.2.1 (intel chip), with pytorch 1.12 and 1.13.

You may want to have a look at the references here:

How to test on MacOS:

import sys
sys.path.append('/path/to/local/d2l-en/pytorch/')
from d2l import torch as d2l
d2l.num_gpus()

1

@d2l-bot
Copy link
Member

d2l-bot commented Mar 9, 2023

Job d2l-en/PR-2453/1 is complete.
Check the results at http://preview.d2l.ai/d2l-en/PR-2453/

@d2l-bot
Copy link
Member

d2l-bot commented Mar 9, 2023

Job d2l-en/PR-2453/2 is complete.
Check the results at http://preview.d2l.ai/d2l-en/PR-2453/

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

Successfully merging this pull request may close these issues.

None yet

2 participants