为算法脚本定义环境 - Amazon Braket

本文属于机器翻译版本。若本译文内容与英语原文存在差异，则一律以英文原文为准。

为算法脚本定义环境

AmazonBraket 支持三种由容器为算法脚本定义的环境：

  • 基础容器（如果未指定，则image_uri为默认容器）

  • 一个装有 Tensorflow 的容器和 PennyLane

  • 带有 PyTorch 和的容器 PennyLane

下表提供了有关容器及其包含的库的详细信息。

亚马逊 Braket 容器
类型 PennyLane 与 TensorFlow PennyLane 与 PyTorch Pennylane

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

其他库

  • 亚马逊 braket-默认模拟器

  • 亚马逊 braket-pennylane-plugin

  • 亚马逊 braket-schemas

  • amazon-braket-sdk

  • ipykernel

  • keras

  • matplotlib

  • 网络x

  • openbabel

  • PennyLane

  • protobuf

  • psi4

  • rsa

  • PennyLane-闪电般的 GPU

  • cuQuan

  • 亚马逊 braket-默认模拟器

  • 亚马逊 braket-pennylane-plugin

  • 亚马逊 braket-schemas

  • amazon-braket-sdk

  • ipykernel

  • keras

  • matplotlib

  • 网络x

  • openbabel

  • PennyLane

  • protobuf

  • psi4

  • rsa

  • PennyLane-闪电般的 GPU

  • cuQuan

  • 亚马逊 braket-默认模拟器

  • 亚马逊 braket-pennylane-plugin

  • 亚马逊 braket-schemas

  • amazon-braket-sdk

  • awscli

  • boto3

  • ipykernel

  • matplotlib

  • 网络x

  • numpy

  • openbabel

  • pandas

  • PennyLane

  • protobuf

  • psi4

  • rsa

  • scipy

您可以在 aws /amazon-braket-conta iners 上查看和访问开源容器定义。选择最符合您的用例的容器。容器必须位于 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 im@@ ag e_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- AWS 区域 2 中检索它们。

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")