-
Notifications
You must be signed in to change notification settings - Fork 429
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
mps device option is selected if you run this repository in MacOS #300
base: main
Are you sure you want to change the base?
Conversation
@cperales did you use custom models for inference? I tried to use your code on Mac with model from README and faced with RuntimeError: It's traced on I'm not sure that it is problem with model, but have no other ideas. Maybe you have? |
I was able to fix this by editing both the
When running on a mac, I take it that we're just trying to change everything to the MPS device, so I simply changed the lines to the following:
After this, I got running the code to work. |
Hi, after all changes I still get "RuntimeError: MPS: Unsupported Border padding mode". Which python version and torch version are you guys using? /edit: fixed it. In case anyone stumbles upon it here; paste this: |
I have been running this code in Ubuntu and Windows (through Flowframes), and I would like to run it in a Macbook 16 that I am using. Reading PyTorch documentation, MPS device enables high-performance training on GPU for MacOS devices with Metal programming framework. Basically, MPS is the analogous to CUDA but in MacOs.
The main problem is that Conv3D is not available yet for MPS, so that part of the code must be run with CPU. The environment variable PYTORCH_ENABLE_MPS_FALLBACK must also set to 1, so I included that part too.