sagemaker-python-sdk
sagemaker-python-sdk copied to clipboard
Neuron Image URI config file is out of date
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.
@saimidu Can we have the config updated in SDK?