faiss-gpu的问题
在faiss中使用出现以下问题的:原先的faiss检索方式:dim=768gpu_id=0index = faiss.IndexFlatL2(dim)res = faiss.StandardGpuResources()self.index = faiss.index_cpu_to_gpu(res, gpu_id, index)出现上述问题是IndexFlatL2是暴力检索,数据量比较大时,GPU内存占用比较大,解决方法:dim=768m=8nlist=100d=8
原创
2022-05-16 11:48:05 ·
2111 阅读 ·
0 评论