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

Research direction

Start by reading the sagemaker.image_uris.retrieve(...) entry point and compare its supported framework, framework-version, and Python-version combinations with deep-learning-containers/available_images.md. Define how a parameterless show_images()-style API should represent all available combinations, and consider how completion can be verified against the listed containers without requiring repeated failed retrievals.

Written by the indexing model from the issue text.

Assessment

Tech stack
aws, machine-learning, python
Domain
cloud, machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.