-
Notifications
You must be signed in to change notification settings - Fork 152
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
Could you make the code non-CUDA compliant? #4
Comments
Sorry, I don't have a non-CUDA device. Please delete all the |
Ok I'll try that. Thank you. |
Unfortunately it does not seem to work. I get the same "AssertionError: Torch not compiled with CUDA enabled". Too bad... |
Sorry, I am too busy now. I'll help you later. |
I have adjusted the code to support cpu inferece, you can ref to: |
My teammate found an non-GPU device and developed an non-CUDA demo for you. https://github.com/Canjie-Luo/MORAN_v2/tree/demo_cpu |
Thank you both @Canjie-Luo and @happog! I managed to get MORAN_v2/tree/demo_cpu working. You've been really helpful. |
Hi. It seems that the code is only working with a GPU. I get an error when I try to lauch the demo:
Indeed, the file L37 of
MORAN_v2/models/morn.py
mentionsgrid = torch.from_numpy(grid).type(self.inputDataType).cuda()
which works only if you have a CUDA system with a GPU.
The text was updated successfully, but these errors were encountered: