aws / aws/sagemaker-python-sdk

PyTorch image_uris for graviton is outdated

Ouverte Adaptée aux débutants
#5,532 1 commentaire 1 réaction 0 personnes assignées Voir sur GitHub
Langage dominant
Python
Étoiles
2.3k
Forks
1.3k
Merge moyen
1 j 22 h
PR mergées (30 j)
35

Description

The list of `inference_graviton` image URIs is out-of-date, making it difficult to deploy models on graviton with the latest PyTorch version.

The SDK says 2.4.0 is latest version:
https://github.com/aws/sagemaker-python-sdk/blob/ad190b995c492ce3904a24502fec63e2eeb53abd/sagemaker-core/src/sagemaker/core/image_uri_config/pytorch.json#L1352-L1363

Although 2.6.0 has been available for a year:
- https://github.com/aws/deep-learning-containers/blob/master/available_images.md#sagemaker-framework-arm64graviton-containers-sm-support-only
- https://github.com/aws/deep-learning-containers/commit/b41cd39059d22041a156c4c47dfaf12438d1d796

## Workaround
It's still possible to manually provide the image URI to the ModelBuilder which leads to a working deployment.

Guide de contribution

Ouvrir le guide de contribution

Piste de recherche

Ouvrez sagemaker-core/src/sagemaker/core/image_uri_config/pytorch.json aux entrées inference_graviton citées dans l’issue. Comparez ces URI avec la liste des conteneurs Graviton de SageMaker et le commit lié de Deep Learning Containers, puis mettez à jour l’entrée de version PyTorch obsolète ; le travail est terminé lorsque le SDK expose l’URI d’image 2.6.0 disponible pour les déploiements Graviton.

Rédigé par le modèle d'indexation à partir du texte de l'issue.

Évaluation

Stack technique
aws, python, pytorch
Domaine
cloud, machine-learning
Type d'issue
Bug
Difficulté
2/5
Temps estimé
1-3 heures
Activité
Calme
Clarté
Clairement spécifiée
Accessibilité débutants
74/100

Recevez les nouvelles issues par e-mail

Un résumé court des issues GitHub adaptées aux débutants.