How to train without the cuda extension?

[Geek-lixiang]

I can not use the cuda extension, how can I train the model without it?
Thank you very much!

[Ali Hassani]

Thank you for your interest.
Unfortunately we don't have a cpu version yet. Were you unable to install the extension, or do you not have CUDA?

[AntonyDrovalov]

Hello!
Can I run inference without using a cuda extension? (for example, if I have macos operating system)

[Ali Hassani]

Unfortunately that is not possible at the moment because we don't have a cpu implementation. We plan to release one in the future.

[AntonyDrovalov]

Unfortunately that is not possible at the moment because we don't have a cpu implementation. We plan to release one in the future.

Thanks for reply. Looking forward to this update.

[Geek-lixiang]

Thank you very much for your reply.

[Ali Hassani]

We released the pure PyTorch version (PR #23 ) along with a new version of the kernel.
I'd like to thank @qwopqwop200 for their implementation of our RPB, which was used instead of our own in the PyTorch version.