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

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

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

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

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

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

Amazon Braket コンテナ

タイプ	PennyLane と TensorFlow	PyTorch を使用した PennyLane	ペンニラネ
基本	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	292282985366.dkr.ecr.us-west-2.amazonaws.com/amazon-braket-base-jobs:latest
継承ライブラリ	awscli numpy pandas scipy	awscli numpy pandas scipy
追加のライブラリ	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	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

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

• ベースイメージ image_uri="292282985366.dkr.ecr.us-west-2.amazonaws.com/amazon-braket-base-jobs:1.0-cpu-py39-ubuntu22.04"

• Tensorflow イメージ image_uri="292282985366.dkr.ecr.us-east-1.amazonaws.com/amazon-braket-tensorflow-jobs:2.11.0-gpu-py39-cu112-ubuntu20.04"

• PyTorch イメージ image_uri="292282985366.dkr.ecr.us-west-2.amazonaws.com/amazon-braket-pytorch-jobs:1.13.1-gpu-py39-cu117-ubuntu20.04"

は、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_tf = retrieve_image(Framework.PL_TENSORFLOW, "us-west-2")
image_uri_pytorch = retrieve_image(Framework.PL_PYTORCH, "us-west-2")