aws / aws/sagemaker-python-sdk

Programmatic way to get available training/inference containers.

Open
#4,318 1 comment 0 reactions 0 assignees View on GitHub
component: container type: feature request
Dominant language
Python
Stars
2.3k
Forks
1.3k
Avg merge
1d 22h
Merged PRs (30d)
35

Description

**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.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.