Description
Describe the bug
The config file 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.