Added mps #337
Added mps #337
Conversation
CGCooke
commented
Mar 23, 2023
CGCooke
commented
- Added MPS as a device option.
- Potential for improved performance on devices that support MPS. (Mac M1/M2)
1. Added [MPS](https://developer.apple.com/documentation/metalperformanceshaders) as a device option, to Interacting_with_CLIP.ipynb. 2. Results in a 30% increase in execution speed for the notebook on an Apple M2 Mac Mini.
1. Added MPS as a device to Prompt_Engineering_for_ImageNet.ipynb
|
Thanks @CGCooke for enabling the "mps" backend option. |
|
Is this dead? I'm running into so many mps problems with so many different models - all running on pytorch. But here with CLIP, detectron2 - all seem to be old/stale issues. Trying to understand expectations - is apple not super committed to silicon being a viable option for using widely available open-sourced ML tooling/models? |
@kevflynn we are committed to provide mps backend support on Mac platforms. The issue which is linked has been long solved on PyTorch nightlies. can you please provide all the issues which you are running into and we will priortize it |
