定義演算法指令碼的環境 - Amazon Braket

本文為英文版的機器翻譯版本，如內容有任何歧義或不一致之處，概以英文版為準。

定義演算法指令碼的環境

Amazon Braket 支援由演算法指令碼容器定義的三個環境：

  • 基礎容器 （預設，如果image_uri未指定）

  • 具有 Tensorflow 和 的容器 PennyLane

  • 具有 PyTorch 和 的容器 PennyLane

下表提供有關容器及其包含的程式庫的詳細資訊。

Amazon Braket 容器
Type PennyLane 使用 TensorFlow PennyLane 使用 PyTorch 潘尼蘭

Base

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

  • 角

  • 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

  • 角

  • 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(…​)呼叫來指定容器映像。您可以將其他相依性安裝到您在執行時間選擇的容器 （啟動或執行時間的成本），因為 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"

image-uris 也可以使用 中的 retrieve_image()函數來擷取 Amazon Braket SDK。下列範例示範如何從 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")