-
Notifications
You must be signed in to change notification settings - Fork 872
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
how can id run this project on Mac OS M1, use the GPU via torch "mps" #30
Comments
same question here |
+1 |
Have you solved the problem |
+1 |
I did everything as I can to port the code to mps. Sadly, mps does not implement ten::_fft_r2c at PyTorch 2.0.1. And I got this from console:
|
@yosefl20 thanks for you insights. i'm unable to get it run on my M1 CPU. I set this env variable to 1 ..but still gets the error
any idea how to workaround this? |
I was able to run this on mps. Still slow but significantly better than CPU. Here is the 5 second video result. MacBook Air 2020 M1/16GB RAM/8 Core GPU MPS ===> 29min 36sec Let me know how long yours take. And if you want to try yourself on mps: 1-) face_parsing.py ---> line 44 to --->
2-) inference.py --->
3-)
We do step 3 because there is this one operation is yet to be implemented using mps. It should work. Cheers. |
Thank you for the solution, but I have installed PyTorch 1.9.0, the torch.backends does have "mps", which PyTorch version do you use and does it work for this project , for the guide says , to install PyTorch 1.9.0 |
how can id run this project on Mac OS M1, use the GPU via torch mps
The text was updated successfully, but these errors were encountered: