定義演算法指令碼的環境 - 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 您也可以使用 Amazon Braket SDK 中的 retrieve_image()函數來擷取 。下列範例示範如何從 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")