princeton-vl / lietorch

BSD 3-Clause "New" or "Revised" License
662 stars 46 forks source link

CPU only version? #16

Closed w-hc closed 2 years ago

w-hc commented 2 years ago

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 2 years ago

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