Definición del entorno para el script de algoritmo - Amazon Braket

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Definición del entorno para el script de algoritmo

Amazon Braket admite entornos definidos por contenedores para el script de algoritmo:

  • Un contenedor base (el predeterminado, si no se especifica un image_uri)

  • Un contenedor con CUDA-Q

  • Un contenedor con Tensorflow y PennyLane

  • Un contenedor con PyTorch, y PennyLane CUDA-Q

En la tabla siguiente, se proporcionan detalles sobre los contenedores y las bibliotecas que incluyen.

Contenedores de Amazon Braket
Tipo Base CUDA-Q TensorFlow PyTorch

URI de imagen

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

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

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

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

Bibliotecas heredadas

  • amazon-braket-default-simulator

  • amazon-braket-pennylane-plugin

  • amazon-braket-schemas

  • amazon-braket-sdk

  • awscli

  • botocore

  • boto3

  • dask

  • matplotlib

  • numpy

  • pandas

  • PennyLane

  • PennyLane-Relámpago

  • qiskit-braket-provider

  • solicitudes

  • sagemaker-training

  • scikit-learn

  • scipy

  • awscli

  • numpy

  • pandas

  • scipy

  • awscli

  • numpy

  • pandas

  • scipy

Bibliotecas adicionales

  • 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-solvers

  • 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-GPU Lightning

  • cuQuantum

  • cudaq

  • cudaq-qec

  • cudaq-solvers

Puede ver y acceder a las definiciones de contenedores de código abierto en aws/. amazon-braket-containers Elija el contenedor que mejor se adapte a su caso de uso. Puedes usar cualquiera de las AWS regiones disponibles en Braket (us-east-1, us-west-1, us-west-2, eu-north-1, eu-west-2), pero la región contenedora debe coincidir con la región de tu trabajo híbrido. Especifique la imagen del contenedor al crear un trabajo híbrido añadiendo uno de los tres argumentos siguientes a su llamada a create(…​) en el script del trabajo híbrido. Puede instalar dependencias adicionales en el contenedor que elija en el tiempo de ejecución (al costo del inicio o el tiempo de ejecución), ya que los contenedores de Amazon Braket tienen conectividad a Internet. En el siguiente ejemplo es para la región us-west-2.

  • Imagen base: amazon-braket-base-jobs image_uri="292282985366.dkr. ecr.us-west-2.amazonaws.com /:latest»

  • Imagen CUDA-Q: image_uri="292282985366.dkr. ecr.us-west-2.amazonaws.com /:latest» amazon-braket-cudaq-jobs

  • Imagen de Tensorflow: image_uri="292282985366.dkr. ecr.us-west-2.amazonaws.com /:latest» amazon-braket-tensorflow-jobs

  • PyTorch imagen: image_uri="292282985366.dkr. ecr.us-west-2.amazonaws.com /:latest» amazon-braket-pytorch-jobs

Los image-uris también se pueden recuperar usando la función retrieve_image() en el SDK de Amazon Braket. El siguiente ejemplo muestra cómo recuperarlos del us-west-2 Región de 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")