admin 管理员组文章数量: 887033
在Windows下,安装Faiss的推荐方法是通过conda。
仅 CPU 的 conda 包目前在 Linux、OSX 和 Windows。包含 CPU 和 GPU 索引的 ,在 Linux 系统,适用于各种版本的 CUDA。下面是官网给出的命令,windows的话用cpu版就行。
# CPU-only version
$ conda install -c pytorch faiss-cpu
# GPU(+CPU) version
$ conda install -c pytorch faiss-gpu
# or for a specific CUDA version
$ conda install -c pytorch faiss-gpu cudatoolkit=10.2 # for CUDA 10.2
- 使用conda安装,开始栏搜索Anaconda Prompt
终端输入下面命令
conda install -c pytorch faiss-cpu
输入y
等待完成,会有三个done
然后打开pycharm,导入faiss包,如果不报错那基本上没什么问题了。
运行一下官网上给出的例子
import numpy as np
d = 64 # dimension
nb = 100000 # database size
nq = 10000 # nb of queries
np.random.seed(1234) # make reproducible
xb = np.random.random((nb, d)).astype('float32')
xb[:, 0] += np.arange(nb) / 1000.
xq = np.random.random((nq, d)).astype('float32')
xq[:, 0] += np.arange(nq) / 1000.
import faiss # make faiss available
index = faiss.IndexFlatL2(d) # build the index
print(index.is_trained)
index.add(xb) # add vectors to the index
print(index.ntotal)
k = 4 # we want to see 4 nearest neighbors
D, I = index.search(xb[:5], k) # sanity check
print(I)
print(D)
D, I = index.search(xq, k) # actual search
print(I[:5]) # neighbors of the 5 first queries
print(I[-5:])
控制台会有输出结果,大功告成了~
版权声明:本文标题:windows下faiss安装详细版 内容由网友自发贡献,该文观点仅代表作者本人, 转载请联系作者并注明出处:http://www.freenas.com.cn/jishu/1716033618h663259.html, 本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
发表评论