rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-10-28 14:36:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: use GPU for index creation. (#540)

Browse Source 
Bump lance to 0.8.3 to include GPU training

---------

Co-authored-by: Rob Meng <rob.xu.meng@gmail.com>
This commit is contained in:
Lei Xu
2023-10-05 20:49:00 -07:00
committed by GitHub
co-authored by Rob Meng
parent 88d8d7249e
commit a26c8f3316
8 changed files with 104 additions and 85 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/src/ann_indexes.md
+6
View File
@@ -68,6 +68,12 @@ a single PQ code.
<figcaption>IVF_PQ index with <code>num_partitions=2, num_sub_vectors=4</code></figcaption>
</figure>
### Use GPU to build vector index
Lance Python SDK has experimental GPU support for creating IVF index.
You can specify the GPU device to train IVF partitions via
- **accelerator**: Specify to `"cuda"`` to enable GPU training.
## Querying an ANN Index