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
Assessment
This issue has not been assessed yet.