ImageUris
- class braket.jobs.image_uris.Framework(*values)[source]
Bases:
StrEnumSupported Frameworks for pre-built containers
- BASE = 'BASE'
- PL_TENSORFLOW = 'PL_TENSORFLOW'
- PL_PYTORCH = 'PL_PYTORCH'
- CUDAQ = 'CUDAQ'
- braket.jobs.image_uris.built_in_images(region)[source]
Checks a region for built in Braket images.
- Parameters:
region (str) – The AWS region to check for images
- Return type:
set[str]- Returns:
set[str] – returns a set of built images
- braket.jobs.image_uris.retrieve_image(framework, region)[source]
Retrieves the ECR URI for the Docker image matching the specified arguments.
- Parameters:
framework (Framework) – The name of the framework.
region (str) – The AWS region for the Docker image.
- Return type:
str- Returns:
str – The ECR URI for the corresponding Amazon Braket Docker image.
- Raises:
ValueError – If any of the supplied values are invalid or the combination of inputs specified is not supported.