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

Posted 2023-03-16

【中文标题】FAISS-GPU 不能与 GPU 版本 30xx 一起使用

【英文标题】：FAISS-GPU cannot be used with GPU version 30xx

【发布时间】：2022-01-09 11:42:03

【问题描述】：

我正在编写 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

【问题讨论】：

请提供足够的代码，以便其他人更好地理解或重现问题。

【参考方案1】：

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

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

conda install -c anaconda pytorch-gpu

【讨论】：

正如目前所写，您的答案尚不清楚。请edit 添加其他详细信息，以帮助其他人了解这如何解决所提出的问题。你可以找到更多关于如何写好答案的信息in the help center。