princeton-vl/lietorch

CPU only version?

Closed this issue · 1 comments

w-hc commented

Hi thanks for the great work.
Is it possible to compile the library for CPU-only use? Do you have some simple instructions on how to modify the setup.py for this purpose?
Thanks in advance.

w-hc commented

Here is my CPU-only fork
https://github.com/w-hc/lietorch
It's a band-aid, but good for the moment.