FAISS-GPU 1.7.1在RTX30xx系列显卡运行异常问题求助

核心原因 #

Faiss 1.7.1版本预编译包默认仅适配图灵架构（RTX 20xx系列及更早显卡），未包含RTX 30xx系列所属安培架构的兼容代码，调用GPU时会出现架构不匹配导致的底层死锁，无法通过中断信号终止。

可行解决方案 #

• 方案1：升级适配安培架构的Faiss版本（推荐）
  1. 卸载现有版本：pip uninstall faiss-gpu -y
  2. 安装1.7.2及以上适配CUDA 11.x的版本：pip install faiss-gpu==1.7.2

  注意：RTX 30系显卡最低要求CUDA版本≥11.1，NVIDIA驱动版本≥455.23，需提前确认环境匹配。

• 方案2：conda环境安装适配组合
  直接安装经过验证的兼容套装：conda install -c pytorch faiss-gpu=1.7.2 cudatoolkit=11.3
• 方案3：自行编译适配参数
  若需要自定义编译Faiss，cmake阶段添加架构指定参数：-DCMAKE_CUDA_ARCHITECTURES="80;86"，其中86对应RTX 30系显卡的计算能力。
• 方案4：临时兼容方案
  若暂时无法升级版本，在Python代码最开头添加环境变量配置，强制CUDA即时编译适配当前显卡架构的二进制文件：

  import os
  os.environ["CUDA_FORCE_PTX_JIT"] = "1"
  

  该方案会导致首次启动程序速度变慢，仅适合临时测试使用。

验证建议 #

修复后先运行单GPU测试用例确认稳定性，再切换至多GPU运行场景，避免多卡资源冲突引入额外异常。

内容的提问来源于stack exchange，提问作者alphazero2003