aws / aws/sagemaker-python-sdk

Neuron Image URI config file is out of date

Open
#3,803 1 comment 0 reactions 0 assignees View on GitHub
component: image uri component: neo type: bug type: config missing
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.