启用 GPU 加速 - 亚马逊 OpenSearch 服务
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

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

启用 GPU 加速

使用、或 SDK 创建或更新 OpenSearch 域名或 OpenSearch 无服务器集合时Amazon Web Services 管理控制台,Amazon CLI您可以启用 GPU 加速。Amazon

在网域或集合上启用 GPU 加速后,默认情况下,所有索引都将启用此功能。如果您需要在索引级别禁用此功能,请参阅创建 GPU 加速的向量索引

控制台

以下过程使用无服务器管理控制台为 OpenSearch 域和 OpenSearch 无服务器集合启用 GPU 加速。 OpenSearch

Create new domain

要创建启用了 GPU 加速的 OpenSearch 域,请参阅。 创建 OpenSearch 服务域

Edit existing domain
  1. 打开OpenSearch 服务管理控制台。

  2. 在导航窗格中,选择

  3. 选择您的域名以打开域名详情页面。

  4. 选择操作,然后选择编辑域名

  5. 高级功能部分中,选择启用 GPU 加速。启用此功能后,您的矢量索引操作将加速

  6. 选择保存更改

Create new collection

要创建启用了 GPU 加速的 OpenSearch 无服务器集合,请参阅。教程:开始使用 Amazon OpenSearch Serverless在创建集合期间,请确保选择矢量搜索集合类型并在矢量搜索配置中启用 GPU 加速。

Edit existing collection
注意

控制台目前不支持通过控制台更新 OpenSearch 无服务器集合上的 GPU 加速状态。但是,即使无法更新集合级别的状态,也可以指定索引设置为特定索引启用 GPU 加速。有关索引级别创建 GPU 加速的向量索引的配置,请参见。如果为索引启用 GPU 加速失败,请联系。Amazon Web Services 支持

Amazon CLI

Create new domain

以下Amazon CLI示例在美国东部(弗吉尼亚北部)创建了一个启用 GPU 加速的 OpenSearch 域。用text您自己的配置替换。

aws opensearch create-domain \ --domain-name my-domain \ --engine-version OpenSearch_3.1 \ --cluster-config InstanceType=r6g.xlarge.search,\ InstanceCount=1,\ DedicatedMasterEnabled=true,\ DedicatedMasterCount=3,\ DedicatedMasterType=m6g.large.search \ --ebs-options "EBSEnabled=true,\ VolumeType=gp3,\ VolumeSize=2000" \ --encryption-at-rest-options '{"Enabled":true}' \ --aiml-options '{"ServerlessVectorAcceleration": {"Enabled": true}}' \ --node-to-node-encryption-options '{"Enabled":true}' \ --domain-endpoint-options '{"EnforceHTTPS":true,\ "TLSSecurityPolicy":"Policy-Min-TLS-1-0-2019-07"}' \ --access-policies '{"Version": "2012-10-17", "Statement": [{ "Effect": "Allow", "Principal": {"AWS": "*"}, "Action": "es:*", "Resource": "arn:aws:es:us-east-1:123456789012:domain/my-domain/*" }]}' \ --advanced-security-options '{ "Enabled":true, "InternalUserDatabaseEnabled":true, "MasterUserOptions": { "MasterUserName":"USER_NAME", "MasterUserPassword":"PASSWORD" }}' \ --region us-east-1
Edit existing domain

以下Amazon CLI示例为现有域启用 GPU 加速。 OpenSearch用text您自己的配置替换。

aws opensearch update-domain-config \ --domain-name my-domain \ --cluster-config InstanceType=r7g.16xlarge.search,InstanceCount=3 \ --aiml-options '{"ServerlessVectorAcceleration": {"Enabled": true}}'
Create new collection

以下Amazon CLI示例在美国东部(弗吉尼亚北部)创建了一个启用 GPU 加速的 OpenSearch 无服务器集合。用text您自己的配置替换。

aws opensearchserverless create-collection \ --name "my-collection" \ --type "VECTORSEARCH" \ --description "My vector collection with GPU acceleration" \ --vector-options '{"ServerlessVectorAcceleration": "ENABLED"}' \ --region us-east-1
Edit existing collection
注意

控制台目前不支持通过控制台更新 OpenSearch 无服务器集合上的 GPU 加速状态。但是,即使无法更新集合级别的状态,也可以指定索引设置为特定索引启用 GPU 加速。有关索引级别创建 GPU 加速的向量索引的配置,请参见。如果为索引启用 GPU 加速失败,请联系。Amazon Web Services 支持

以下Amazon CLI示例为现有的 OpenSearch 无服务器集合启用 GPU 加速。用text您自己的配置替换。

aws opensearchserverless update-collection \ --id collection-id \ --type VECTORSEARCH \ --description "Collection with GPU acceleration enabled"