在自己的容器中运行 Braket 混合作业 - Amazon Braket

在自己的容器中运行 Braket 混合作业

要使用自己的容器创建混合作业，请使用指定的参数 image_uri 调用 AwsQuantumJob.create()。您可以使用 QPU、按需模拟器，也可以在 Braket Hybrid Jobs 提供的经典处理器上对代码进行本地运行。我们建议先在 SV1、DM1 或 TN1 等模拟器上测试代码，然后再在真正的 QPU 上运行。

要在经典处理器上运行代码，请通过更新 InstanceConfig 来指定您使用的 instanceType 和 instanceCount。请注意，如果您指定 instance_count > 1，则需要确保您的代码可以在多个主机上运行。您可以选择的实例数量上限为 5。例如：

job = AwsQuantumJob.create(
    source_module="source_dir",
    entry_point="source_dir.algorithm_script:start_here",
    image_uri="111122223333.dkr.ecr.us-west-2.amazonaws.com/my-byoc-container:latest",
    instance_config=InstanceConfig(instanceType="ml.p3.8xlarge", instanceCount=3),
    device="local:braket/braket.local.qubit",
    # ...)
注意

使用设备 ARN 跟踪您用作混合作业元数据的模拟器。可接受的值必须遵循格式 device = "local:<provider>/<simulator_name>"。请记住 <provider> 和 <simulator_name> 必须仅包含字母、数字、_、- 和 .。字符串大小限制为 256 个字符。

如果您计划使用 BYOC，但不使用 Braket SDK 创建量子任务，则应将环境变量 AMZN_BRAKET_JOB_TOKEN 的值传递给 CreateQuantumTask 请求中的 jobToken 参数。如果您不这样做，量子任务就不会获得优先级，而是作为常规的独立量子任务计费。