可以试试faiss，，安装pip install faiss-cpu或者faiss-gpu
import numpy as np
import faiss
# 创建一些随机向量
d = 64 # 向量维度
nb = 1000 # 向量数量
np.random.seed(1234) # 设置随机种子
xb = np.random.random((nb, d)).astype('float32')
# 创建索引
index = faiss.IndexFlatL2(d) # 使用 L2 距离
index.add(xb) # 添加向量到索引
# 查询
nq = 5 # 查询的向量数量
xq = np.random.random((nq, d)).astype('float32') # 生成查询向量
D, I = index.search(xq, k=10) # 搜索最近的 10 个邻居
print(I) # 输出最近邻的索引还有annoy，，安装pip install annoyfrom annoy import AnnoyIndex
f = 40 # 向量维度
t = AnnoyIndex(f, 'angular') # 使用角度距离
# 添加向量
for i in range(1000):
v = [random.gauss(0, 1) for z in range(f)]
t.add_item(i, v)
t.build(10) # 建立树
t.save('test.ann') # 保存索引
# 查询
print(t.get_nns_by_item(0, 10)) # 查找与第 0 个向量最近的 10 个向量 网友回复
我知道答案，我要回答
私活外包
最新提问
有没有不依赖embedding向量的RAG技术？
有没有支持实时打断语音通话并后台帮你执行任何的ai模型？
开源ai大模型文件格式GGUF、MLX、Safetensors、 ONNX 有什么区别？
出海挣钱支付收款PayPal、Wise 、PingPong、Stripe如何选择？
如何实现类似google的图片隐形水印添加和识别技术？
linux上如何运行任意windows程序？
ai能写出比黑客还厉害的零日漏洞等攻击工具攻击任意软件系统工程？
js如何获取浏览器的音频上下文指纹、Canvas指纹、WebGL渲染特征？
为啥ai开始抛弃markdown文本，重新偏好html文本了？
网站有没有办法鉴别访问请求是由ai操控chrome-devtools-mcp发出的？
