aws / aws/sagemaker-python-sdk
Neuron Image URI config file is out of date
- Dominant language
- Python
- Stars
- 2.3k
- Forks
- 1.3k
- Avg merge
- 1d 22h
- Merged PRs (30d)
- 35
Description
**Describe the bug**
The config file [src/sagemaker/image_uri_config/pytorch-neuron.json](https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/image_uri_config/pytorch-neuron.json) is out of date as it is missing many of the pre-built containers that are described here: https://github.com/aws/deep-learning-containers/blob/master/available_images.md#neuron-containers
Also, is there a TensorFlow config file for Neuron ?
**To reproduce**
A clear, step-by-step set of instructions to reproduce the bug.
**Expected behavior**
A clear and concise description of what you expected to happen.
Update the JSON file with the newest container images for Neuron.
**Screenshots or logs**
If applicable, add screenshots or logs to help explain your problem.
**System information**
A description of your system. Please provide:
- **SageMaker Python SDK version**:
- **Framework name (eg. PyTorch) or algorithm (eg. KMeans)**:
- **Framework version**:
- **Python version**:
- **CPU or GPU**:
- **Custom Docker image (Y/N)**:
**Additional context**
Add any other context about the problem here.
Contributor guide
Research direction
Compare src/sagemaker/image_uri_config/pytorch-neuron.json with the Neuron container entries in available_images.md. Check whether the repository contains a TensorFlow Neuron configuration file, then update or add the relevant configuration based on the documented images. Done means the SDK configuration includes the missing current Neuron containers.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- aws, python, pytorch, tensorflow
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 2/5
- Estimated time
- 1-3 hours
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100