-
Notifications
You must be signed in to change notification settings - Fork 1.3k
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
ns-train splatfacto fails to run on Apple Silicon, nerfacto works ok #3290
Comments
Splatfacto requires CUDA for training right now. If you are very keen and willing to hack away, it is in theory possible to train with cpu torch right now using the _torch_impl from the cuda backend but its not supported out of the box. |
Thanks for the quick response @maturk ! I see, support for MPS across the ecosystem seems to be evolving quickly now, there is a lot of active development, but there is still a lot of ground to cover. One last question realted to this topic. There are several improvements for MPS on the latest PyTorch versions, but when I tried to upgrade it to latest, I could not longer run ns-train. Do you think it is hard to resolve those conflicts? The error thrown is a bit cryptic:
Without the fallback flag:
|
Original question answered, no need to keep open. |
Describe the bug
I have an M3 and could successfully install nerfstudio, and (albeit slowly) train my first model using nerfacto. The problem is that when I try to use splatfacto, it exceptions out, saying:
Naturally my torch does not have CUDA enabled since there is no Nvidia GPU. I tried with --machine.device-type=cpu, with the same result.
Command that works:
ns-train nerfacto --machine.device-type=mps --max-num-iterations 2000 --data /Users/user/dev/oss/nerfstudio/data/nerfstudio/desk
Command that produces the above exception:
ns-train splatfacto --machine.device-type=mps --max-num-iterations 2000 --data /Users/user/dev/oss/nerfstudio/data/nerfstudio/desk
Additional context
nerfstudio 1.1.3
torch 2.1.2
torchvision 0.16.2
Thanks!
The text was updated successfully, but these errors were encountered: