翻訳は機械翻訳により提供されています。提供された翻訳内容と英語版の間で齟齬、不一致または矛盾がある場合、英語版が優先します。

ハイブリッドジョブインスタンスの設定

アルゴリズムによっては、要件が異なる場合があります。デフォルトでは、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.
    ...
    ),

埋め込みシミュレーションを実行する場合に、デバイス設定でローカルデバイスが指定されているときは、instanceCount を指定して 1 より大きな値に設定することで、InstanceConfig で複数のインスタンスを追加でリクエストできます。上限は 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
    ...
    ),

複数のインスタンスを使用する場合は、データ並列機能を使用してハイブリッドジョブを分散することを検討してください。これを表示する方法の詳細については、ノートブックの例である「Parallelize training for QML」を参照してください。

次の 3 つの表は、スタンダードインスタンス、ハイパフォーマンスインスタンス、GPU 高速化インスタンスのぞれぞれで使用可能なインスタンスタイプと仕様の一覧です。

各インスタンスは、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 バケットの場所を指定する機能など、柔軟性が向上します。デフォルトでは、Amazon S3 バケットの場所 "amazon-braket-{id}-{region}" が AwsSession に事前設定されています。ただし、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
)