アルゴリズムスクリプトの環境を定義する - Amazon Braket

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

アルゴリズムスクリプトの環境を定義する

Amazon Braket は、アルゴリズムスクリプトのコンテナによって定義された環境をサポートします。

  • ベースコンテナ ( が指定されていない場合image_uriはデフォルト）

  • CUDA-Q を使用するコンテナ

  • Tensorflow と PennyLane を使用するコンテナ

  • PyTorch、PennyLane、CUDA-Q を含むコンテナ

次の表に、コンテナとそのコンテナに含まれるライブラリの詳細を示します。

Amazon Braket コンテナ
タイプ 基本 CUDA-Q TensorFlow PyTorch

イメージ URI

292282985366.dkr.ecr.us-west-2.amazonaws.com/amazon-braket-base-jobs:latest

292282985366.dkr.ecr.us-west-2.amazonaws.com/amazon-braket-cudaq-jobs:latest

292282985366.dkr.ecr.us-east-1.amazonaws.com/amazon-braket-tensorflow-jobs:latest

292282985366.dkr.ecr.us-west-2.amazonaws.com/amazon-braket-pytorch-jobs:latest

継承ライブラリ

  • amazon-braket-default-simulator

  • amazon-braket-PennyLane-plugin

  • amazon-braket-schemas

  • amazon-braket-sdk

  • awscli

  • botocore

  • boto3

  • ダスク

  • matplotlib

  • numpy

  • pandas

  • PennyLane

  • PennyLane-Lightning

  • qiskit-braket-provider

  • リクエスト

  • sagemaker-training

  • scikit-learn

  • scipy

  • awscli

  • numpy

  • pandas

  • scipy

  • awscli

  • numpy

  • pandas

  • scipy

追加のライブラリ

  • amazon-braket-default-simulator

  • amazon-braket-PennyLane-plugin

  • amazon-braket-schemas

  • amazon-braket-sdk

  • awscli

  • boto3

  • ipykernel

  • matplotlib

  • networkx

  • numpy

  • openbabel

  • pandas

  • PennyLane

  • protobuf

  • psi4

  • rsa

  • scipy

  • cudaq

  • cudaq-qec

  • cudaq-solver

  • amazon-braket-default-simulator

  • amazon-braket-PennyLane-plugin

  • amazon-braket-schemas

  • amazon-braket-sdk

  • ipykernel

  • keras

  • matplotlib

  • networkx

  • openbabel

  • PennyLane

  • protobuf

  • psi4

  • rsa

  • PennyLane-Lightning-gpu

  • cuQuantum

  • amazon-braket-default-simulator

  • amazon-braket-PennyLane-plugin

  • amazon-braket-schemas

  • amazon-braket-sdk

  • ipykernel

  • keras

  • matplotlib

  • networkx

  • openbabel

  • PennyLane

  • protobuf

  • psi4

  • rsa

  • PennyLane-Lightning-gpu

  • cuQuantum

  • cudaq

  • cudaq-qec

  • cudaq-solver

オープンソースのコンテナ定義は、aws/amazon-braket-containers で表示およびアクセスできます。ユースケースに一致するコンテナを選択します。Braket で使用可能な AWS リージョン (us-east-1、us-west-1、us-west-2、eu-north-1、eu-west-2) のいずれかを使用できますが、コンテナリージョンはハイブリッドジョブのリージョンと一致する必要があります。ハイブリッドジョブスクリプトのcreate(…​)呼び出しに次の 3 つの引数のいずれかを追加して、ハイブリッドジョブを作成するときにコンテナイメージを指定します。Amazon Braket コンテナにはインターネット接続があるため、ランタイム時に (スタートアップまたはランタイムのコストで) 選択したコンテナに追加の依存関係をインストールできます。次の例は、us-west-2 リージョン用です。

  • ベースイメージ： image_uri"292282985366.dkr.ecr.us-west-2.amazonaws.com/amazon-braket-base-jobs:latest"

  • CUDA-Q イメージ： image_uri"292282985366.dkr.ecr.us-west-2.amazonaws.com/amazon-braket-cudaq-jobs:latest"

  • Tensorflow イメージ： image_uri"292282985366.dkr.ecr.us-west-2.amazonaws.com/amazon-braket-tensorflow-jobs:latest"

  • PyTorch イメージ： image_uri"292282985366.dkr.ecr.us-west-2.amazonaws.com/amazon-braket-pytorch-jobs:latest"

は、AmazonBraket SDK の retrieve_image()関数を使用して取得image-urisすることもできます。次の例は、us-west-2 からそれらを取得する方法を示しています AWS リージョン。

from braket.jobs.image_uris import retrieve_image, Framework

image_uri_base = retrieve_image(Framework.BASE, "us-west-2")
image_uri_cudaq = retrieve_image(Framework.CUDAQ, "us-west-2")
image_uri_tf = retrieve_image(Framework.PL_TENSORFLOW, "us-west-2")
image_uri_pytorch = retrieve_image(Framework.PL_PYTORCH, "us-west-2")