sagemaker-python-sdk icon indicating copy to clipboard operation
sagemaker-python-sdk copied to clipboard

Programmatic way to get available training/inference containers.

Open Alex-Wenner-FHR opened this issue 1 year ago • 1 comments

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.

Alex-Wenner-FHR avatar Dec 12 '23 17:12 Alex-Wenner-FHR

Related Issue: https://github.com/aws/deep-learning-containers/issues/2732

Alex-Wenner-FHR avatar Dec 13 '23 14:12 Alex-Wenner-FHR