创建 GPU 加速的向量索引 - 亚马逊 OpenSearch 服务

本文属于机器翻译版本。若本译文内容与英语原文存在差异,则一律以英文原文为准。

创建 GPU 加速的向量索引

在域名或集合上启用 GPU 加速后,创建可以利用 GPU 处理能力的矢量索引。

注意

当您创建启用了 GPU 加速的域时,默认index.knn.remote_index_build.enabled设置为true该设置。创建索引时无需显式设置此设置。对于集合,您必须为此设置明确指定一个值。

Creating index with GPU-acceleration

以下示例创建了针对 GPU 处理进行了优化的向量索引。该索引存储 768 维向量(常见于文本嵌入)。

PUT my-vector-index { "settings": { "index.knn": true, "index.knn.remote_index_build.enabled": true }, "mappings": { "properties": { "vector_field": { "type": "knn_vector", "dimension": 768 }, "text": { "type": "text" } } } }

关键配置元素:

  • "index.knn": true-启用 k 最近邻功能

  • "index.knn.remote_index_build.enabled": true-为该索引启用 GPU 处理。如果域已启用 GPU 加速,则不指定此设置默认为。true对于集合,您必须为此设置明确指定一个值。

  • "dimension": 768-指定矢量大小(根据您的嵌入模型进行调整)

Creating index without GPU-acceleration

以下示例创建了一个禁用 GPU 处理的向量索引。该索引存储 768 维向量(常见于文本嵌入)。

PUT my-vector-index { "settings": { "index.knn": true, "index.knn.remote_index_build.enabled": false }, "mappings": { "properties": { "vector_field": { "type": "knn_vector", "dimension": 768 }, "text": { "type": "text" } } } }