python - FAISS-GPU 不能与 GPU 版本 30xx 一起使用

Question

我正在编写 Python 源代码并使用 Faiss。我可以在 CPU、20xx GPU 上使用 Faiss，例如：RTX 2080Ti，... 但是，当我在 RTX 3060、RTX 3070 等设备上运行时，系统会死机，我无法通过 Ctrl + C 杀死程序。这是我使用的源代码：https ://github.com/facebookresearch/faiss/blob/main/tutorial/python/5-Multiple-GPUs.py

如果有人遇到此错误并成功修复它，请与我分享如何解决。

操作系统：Ubuntu 20.04

Faiss版本：发布1.7.1

运行在：GPU + Python

Answer 1

我使用了 Anaconda，我安装了波纹管，它起作用了。如果你使用 Docker，请安装 Miniconda。

conda install faiss-gpu cudatoolkit=11.1 -c pytorch-gpu

conda install -c anaconda pytorch-gpu