facebookresearch/faiss

Memory requirement for GPU KMeans

VoVAllen opened this issue · 0 comments

Hi, I want to use Faiss and GPU to perform KMeans on my own data and get the centroids. What are the gpu and cpu memory requirements for Faiss GPU? I hope to cluster 300GB of data with an L4 (24GB) GPU.

Also could you also suggest some code for me to learn from? There're multiple gpu examples and some seems outdated.