配置您的混合作业实例 - Amazon Braket

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

配置您的混合作业实例

根据您的算法,您可能有不同的要求。默认情况下,Amazon Braket 会在 ml.m5.large 实例上运行您的算法脚本。但是,在创建混合作业时,您可以使用以下导入和配置参数自定义此实例类型。

from braket.jobs.config import InstanceConfig job = AwsQuantumJob.create( ... instance_config=InstanceConfig(instanceType="ml.g4dn.xlarge"), # Use NVIDIA T4 instance with 4 GPUs. ... ),

如果您正在运行嵌入式模拟并在设备配置中指定了本地设备,则还可以通过 InstanceConfig 指定 instanceCount 并将其设置为大于一台来请求多个实例。上限为 5。例如,您可以按如下方式选择 3 个实例。

from braket.jobs.config import InstanceConfig job = AwsQuantumJob.create( ... instance_config=InstanceConfig(instanceType="ml.g4dn.xlarge", instanceCount=3), # Use 3 NVIDIA T4 instances ... ),

当您使用多个实例时,请考虑使用数据 parallel 功能分配混合作业。有关如何查看 QML 的 Parallelize 训练示例的更多详细信息,请参阅以下示例笔记本。

以下三个表格列出了标准、高性能和 GPU 加速实例的可用实例类型及规格。

注意

要查看混合任务的默认传统计算实例配额,请参阅 Amazon Braket 配额页面

标准实例 vCPU 内存(GiB)

ml.m5.large(默认)

4

16

ml.m5.xlarge

4

16

ml.m5.2xlarge

8

32

ml.m5.4xlarge

16

64

ml.m5.12xlarge

48

192

ml.m5.24xlarge

96

384

高性能实例 vCPU 内存(GiB)

ml.c5.xlarge

4

8

ml.c5.2xlarge

8

16

ml.c5.4xlarge

16

32

ml.c5.9xlarge

36

72

ml.c5.18xlarge

72

144

ml.c5n.xlarge

4

10.5

ml.c5n.2xlarge

8

21

ml.c5n.4xlarge

16

32

ml.c5n.9xlarge

36

72

ml.c5n.18xlarge

72

192

GPU 加速实例 GPUs vCPU 内存(GiB) GPU 内存 (GiB)

ml.p4d.24xlarge

8

96

1152

320

ml.g4dn.xlarge

1

4

16

16

ml.g4dn.2xlarge

1

8

32

16

ml.g4dn.4xlarge

1

16

64

16

ml.g4dn.8xlarge

1

32

128

16

ml.g4dn.12xlarge

4

48

192

64

ml.g4dn.16xlarge

1

64

256

16

每个实例使用 30 GB 的数据存储(SSD)的默认配置。但是,您可以按照与配置 instanceType 相同的方式调整存储。以下示例说明如何将总存储空间增加到 50 GB。

from braket.jobs.config import InstanceConfig job = AwsQuantumJob.create( ... instance_config=InstanceConfig( instanceType="ml.g4dn.xlarge", volumeSizeInGb=50, ), ... ),

AwsSession 中配置默认存储桶

使用自己的 AwsSession 实例可以提高灵活性,比如能够为默认 Amazon S3 存储桶指定自定义位置。默认情况下,预先配置的 AwsSession 的 Amazon S3 存储桶位置为 "amazon-braket-{id}-{region}"。但是,在创建 AwsSession 时,您可以选择覆盖默认 Amazon S3 的存储桶位置。用户可以选择将 AwsSession 对象传递到 AwsQuantumJob.create() 方法中,提供 aws_session 参数,如以下代码示例所示。

aws_session = AwsSession(default_bucket="amazon-braket-s3-demo-bucket") # Then you can use that AwsSession when creating a hybrid job job = AwsQuantumJob.create( ... aws_session=aws_session )