独自のコンテナでの Braket ハイブリッドジョブの実行 - Amazon Braket

独自のコンテナでの Braket ハイブリッドジョブの実行

独自のコンテナでハイブリッドジョブを作成するには、指定された image_uri 引数を使用して、AwsQuantumJob.create() を呼び出します。QPU かオンデマンドシミュレーターを使用することも、Braket Hybrid Jobs で使用できる古典的プロセッサでコードをローカルに実行することもできます。実際の QPU で実行する前に、SV1、DM1、TN1 などのシミュレーターでコードをテストすることをお勧めします。

古典的プロセッサでコードを実行するには、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 パラメータに渡す必要があります。そうしないと、量子タスクは優先されず、通常のスタンドアロン量子タスクとして請求されます。