[feature-request] GPU PyTorch Python310 with CUDA on Graviton

[yahavb]

Checklist

• [x ] I've prepended issue tag with type of change: [feature]
• [x ] (If applicable) I've documented below the DLC image/dockerfile this relates to
• [ x] (If applicable) I've documented the tests I've run on the DLC image
• [ x] I'm using an existing DLC image listed here: https://docs.aws.amazon.com/deep-learning-containers/latest/devguide/deep-learning-containers-images.html
• [x ] I've built my own container based off DLC (and I've attached the code used to build my own image)

Concise Description: Is there a DLC that supports GPU PyTorch Python310 with CUDA on Graviton like
763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:2.0.1-gpu-py310-cu118-ubuntu20.04-ec2

DLC image/dockerfile:

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 [...]

Describe the solution you'd like
A clear and concise description of what you want to happen.

Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.

Additional context
Add any other context or screenshots about the feature request here.

[tejaschumbalkar]

Hello@yahavb

Thank you for submitting the feature request. We will discuss and prioritize GPU support for Graviton DLC. Unfortunately I cannot comment on a timeline for this feature. We will update the public documentation and GitHub release notes once the feature is released.