faiss/python
Andrei Ruslantsev 053dc46036
Add support to use selected GPUs by id (#1065)
* Add support to use selected GPUs by id

One can pass parameter gpus to index_cpu_to_gpus_list to use only selected GPU's, i.e. if we pass gpus=[1, 3, 4], we will use 1-st, 3-rd and 4-th graphical cards

* merge functions to avoid code duplication

* use first gpus if none passed
2020-02-25 10:17:15 +01:00
..
Makefile
faiss.py Add support to use selected GPUs by id (#1065) 2020-02-25 10:17:15 +01:00
setup.py
swigfaiss.swig