スクリプトを実行するようにハイブリッドジョブインスタンスを設定する - Amazon Braket

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

スクリプトを実行するようにハイブリッドジョブインスタンスを設定する

アルゴリズムによっては、要件が異なる場合があります。デフォルトでは、Amazon Braket はアルゴリズムスクリプトをml.m5.large インスタンスで実行します。ただし、次のインポートおよび設定引数を使用してハイブリッドジョブを作成するときに、このインスタンスタイプをカスタマイズできます。

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

埋め込みシミュレーションを実行していて、デバイス設定でローカルデバイスを指定している場合は、 を指定instanceCountして 1 つ以上のインスタンスを追加InstanceConfigでリクエストできます。上限は 5 です。たとえば、次のように 3 つのインスタンスを選択できます。

from braket.jobs.config import InstanceConfig job = AwsQuantumJob.create( ... instance_config=InstanceConfig(instanceType="ml.p3.8xlarge", instanceCount=3), # Use 3 NVIDIA Tesla V100 ... ),

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

次の 3 つの表に、標準インスタンス、高性能インスタンス、GPU アクセラレーションインスタンスで使用できるインスタンスタイプと仕様を示します。

注記

ハイブリッドジョブのデフォルトの従来のコンピューティングインスタンスクォータを表示するには、Amazon Braket Quotas」ページを参照してください。

スタンダードインスタンス 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.c5d.4xlarge

16

32

ml.c5d.9xlarge

36

72

ml.c5d.18xlarge

72

192

GPU アクセラレーションインスタンス GPU vCPU メモリ (GiB) GPU メモリ (GiB)

ml.p3.2xlarge

1

8

61

16

ml.p3.8xlarge

4

32

244

64

ml.p3.16xlarge

8

64

488

128

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

注記

p3 インスタンスは us-west-1 では使用できません。ハイブリッドジョブがリクエストされた ML コンピューティングキャパシティをプロビジョニングできない場合は、別のリージョンを使用します。

各インスタンスは、30 GB のデータストレージ (SSD) のデフォルト設定を使用します。ただし、ストレージは、instanceType を設定するのと同じ方法で調整できます。。次の例では、ストレージの合計を 50 GB に増やす方法を示します。

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

AwsSession でデフォルトのバケットを設定します。

独自のAwsSessionインスタンスを使用すると、デフォルトの Amazon S3 バケットのカスタムロケーションを指定する機能など、柔軟性が向上します。デフォルトでは、 の Amazon S3 バケットの場所AwsSessionは事前設定されています"amazon-braket-{id}-{region}"。ただし、 の作成時にデフォルトの Amazon S3 バケットの場所を上書きすることもできますAwsSession。ユーザーは、次のコード例に示すように aws_sessionパラメータを指定することで、オプションで AwsSession オブジェクトを AwsQuantumJob.create()メソッドに渡すことができます。

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 )