Skip to content

Programmatic way to get available training/inference containers. #4318

Open
@Alex-Wenner-FHR

Description

@Alex-Wenner-FHR

Describe the feature you'd like
It would be very helpful to be able to programmatically get all available framerworks, py versions etc, prior to sagemaker.image_uris.retrieve(...).
This would allow people to see what containers are available, without having to dig through repositories and try to find supported images or continuously hit errors until the right permutation of framework, framework version, and py version are achieved.

How would this feature be used? Please describe.
Something like sagemaker.image_uris.show_images() or something like that would be desired. This ideally would take no parameters and show maybe a dictionary of all the frameworks, framework versions, and respective py verisons.

Describe alternatives you've considered
Considering programmatically parsing: https://github.com/aws/deep-learning-containers/blob/master/available_images.md

But this does not feel like a good solution at all.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions