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)

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐