Checklist
Concise Description:
Neuron containers for inference use old SDK (1.19), stale pytorch versions (1.10.2), py3.7 and ubuntu 18.04. Out of sync with current neuron version and neuron training containers.
DLC image/dockerfile:
Pytorch neuron inference container
Is your feature request related to a problem? Please describe.
A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
There is a version mismatch between current neuron SDK, pytorch, python and available container between training and inference.
Describe the solution you'd like
A clear and concise description of what you want to happen.
Update inference container to versions consistent with training container, and current neuron SDK. This is SDK 2.4.0, pytorch 1.11.0, python 3.8 and ubuntu 20.04.
Eg. 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuron:1.11.0-neuron-py38-sdk2.4.0-ubuntu20.04
Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.
N/A
Additional context
Add any other context or screenshots about the feature request here.
N/A
Checklist
Concise Description:
Neuron containers for inference use old SDK (1.19), stale pytorch versions (1.10.2), py3.7 and ubuntu 18.04. Out of sync with current neuron version and neuron training containers.
DLC image/dockerfile:
Pytorch neuron inference container
Is your feature request related to a problem? Please describe.
A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
There is a version mismatch between current neuron SDK, pytorch, python and available container between training and inference.
Describe the solution you'd like
A clear and concise description of what you want to happen.
Update inference container to versions consistent with training container, and current neuron SDK. This is SDK 2.4.0, pytorch 1.11.0, python 3.8 and ubuntu 20.04.
Eg.
763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuron:1.11.0-neuron-py38-sdk2.4.0-ubuntu20.04Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.
N/A
Additional context
Add any other context or screenshots about the feature request here.
N/A