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

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.