Not able to run gpu version on NVIDIA A100/A10

kyamagu / faiss-wheels

Unofficial faiss wheel builder

MIT License

275 stars 35 forks source link

Not able to run gpu version on NVIDIA A100/A10 #97

Closed SandroChen closed 4 months ago

SandroChen commented 4 months ago

version: faiss-gpu 1.7.2

file_content: import faiss gpu_res = faiss.StandardGpuResources() index = faiss.GpuIndexFlatL2(gpu_res, d)

description: After I run the code above, it stuck forever. The same code is workable on NVIDIA V100. Is there any solution? I saw your suggestion in #54 , but the archive seems gone and I cannot get any alternative one.

kyamagu commented 4 months ago

We don't support GPU wheels anymore.

Wheels compatible with A100s are available on the v1.7.3 release page, but no support is planned in the future.

SandroChen commented 4 months ago

Thanks! It works perfectly.