Tentukan lingkungan untuk skrip algoritme Anda - Amazon Braket

Terjemahan disediakan oleh mesin penerjemah. Jika konten terjemahan yang diberikan bertentangan dengan versi bahasa Inggris aslinya, utamakan versi bahasa Inggris.

Tentukan lingkungan untuk skrip algoritme Anda

AmazonBraket mendukung tiga lingkungan yang ditentukan oleh kontainer untuk skrip algoritme Anda:

  • Sebuah wadah dasar (default, jika tidak image_uri ditentukan)

  • Wadah dengan Tensorflow dan PennyLane

  • Sebuah wadah dengan PyTorch dan PennyLane

Tabel berikut memberikan rincian tentang wadah dan pustaka yang disertakan.

Wadah Amazon Braket
Tipe PennyLane dengan TensorFlow PennyLane dengan PyTorch Pennylane

Basis

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

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

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

Perpustakaan yang diwariskan

  • awscli

  • numpy

  • panda

  • scipy

  • awscli

  • numpy

  • panda

  • scipy

Perpustakaan Tambahan

  • amazon-braket-default-simulator

  • amazon-braket-pennylane-plugin

  • amazon-braket-schemas

  • amazon-braket-sdk

  • ipykernel

  • keras

  • matplotlib

  • jaringanx

  • openbabel

  • PennyLane

  • protobuf

  • psi4

  • rsa

  • PennyLane-GPU petir

  • CuQuantum

  • amazon-braket-default-simulator

  • amazon-braket-pennylane-plugin

  • amazon-braket-schemas

  • amazon-braket-sdk

  • ipykernel

  • keras

  • matplotlib

  • jaringanx

  • openbabel

  • PennyLane

  • protobuf

  • psi4

  • rsa

  • PennyLane-GPU petir

  • CuQuantum

  • amazon-braket-default-simulator

  • amazon-braket-pennylane-plugin

  • amazon-braket-schemas

  • amazon-braket-sdk

  • awscli

  • boto3

  • ipykernel

  • matplotlib

  • jaringanx

  • numpy

  • openbabel

  • panda

  • PennyLane

  • protobuf

  • psi4

  • rsa

  • scipy

Anda dapat melihat dan mengakses definisi wadah open source di aws/ amazon-braket-containers. Pilih wadah yang paling cocok dengan kasus penggunaan Anda. Wadah harus berada di Wilayah AWS tempat Anda menjalankan pekerjaan hibrida Anda. Anda menentukan gambar kontainer saat membuat pekerjaan hybrid dengan menambahkan salah satu dari tiga argumen berikut ke create(…​) panggilan Anda dalam skrip pekerjaan hybrid. Anda dapat menginstal dependensi tambahan ke dalam wadah yang Anda pilih saat runtime (dengan biaya startup atau runtime) karena kontainer Amazon Braket memiliki konektivitas internet. Contoh berikut adalah untuk Wilayah us-west-2.

  • Gambar dasar image_uri="292282985366.dkr. ecr.us-west-2.amazonaws.com /:1.0-cpu-py39-ubuntu22.04" amazon-braket-base-jobs

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

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

Ini juga image-uris dapat diambil menggunakan retrieve_image() fungsi di Amazon Braket SDK. Contoh berikut menunjukkan cara mengambilnya dari Wilayah AWS us-west-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")