pygraphistry icon indicating copy to clipboard operation
pygraphistry copied to clipboard

[FEA] faiss-gpu support

Open lmeyerov opened this issue 2 years ago • 1 comments

Is your feature request related to a problem? Please describe.

We are getting some conda version conflicts with faiss-cpu, so switching to pip for now, but it'd probably be even better to support faiss-gpu when available, even if just single-gpu . It looks doable: https://github.com/facebookresearch/faiss/wiki/Running-on-GPUs#in-python-2 .

Describe the solution you'd like

Automatic opt-in to faiss-gpu when available

lmeyerov avatar Mar 29 '23 05:03 lmeyerov

cc @tanmoyio @silkspace

I'll try to test pip install faiss-cpu / #cleanup against the base image for now...

lmeyerov avatar Mar 29 '23 05:03 lmeyerov