1.问题
windows11使用pip install faiss-gpu安装FAISS向量数据库失败。
2.原因
FAISS安装可以通过pip install faiss-gpu和pip install faiss-cpu两种。
faiss-cpu支持Linux(x86_64 和 arm64)、OSX(仅限 arm64)和 Windows(x86_64)
faiss-gpu 包含 CPU 和 GPU 索引,在 Linux 上可用(仅限 x86_64),适用于 CUDA 11.4 和 12.1
3.安装
conda安装
# CPU-only version
$ conda install -c pytorch faiss-cpu=1.8.0
# GPU(+CPU) version
$ conda install -c pytorch -c nvidia faiss-gpu=1.8.0
# GPU(+CPU) version with NVIDIA RAFT
$ conda install -c pytorch -c nvidia -c rapidsai -c conda-forge faiss-gpu-raft=1.8.0
windows11安装FAISS
pip install faiss-cpu
详情见:faiss/INSTALL.md at main · facebookresearch/faiss (github.com)