-
-
Notifications
You must be signed in to change notification settings - Fork 85
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
M1 GPU Support (MPS) #29
Comments
See PR #30 Actually MacOS 13.3 is required. |
In addition to this if anyone is using mokuro I also have the following patches on comic text detector: general.py Line 19 CUDA = True if torch.cuda.is_available() else False
DEVICE = 'cuda' if CUDA else 'cpu' became DEVICE = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu" basemodel.py Line 13 CUDA = True if torch.cuda.is_available() else False
DEVICE = 'cuda' if CUDA else 'cpu' became DEVICE = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu" |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add MPS device type so that M1/M2 gpus can be utilised
See: https://pytorch.org/docs/stable/notes/mps.html
The text was updated successfully, but these errors were encountered: