aws / aws/sagemaker-python-sdk

PyTorch image_uris for graviton is outdated

オープン 初心者向け
#5,532 コメント 1 件 リアクション 1 件 担当者 0 名 GitHub で見る
主要言語
Python
スター
2.3k
フォーク
1.3k
平均マージ
1日 22時間
マージ済み PR(30日)
35

説明

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.

コントリビューションガイド

コントリビューションガイドを開く

調査の方向性

sagemaker-core/src/sagemaker/core/image_uri_config/pytorch.json を、issue で引用されている inference_graviton エントリの箇所で開きます。これらの URI を SageMaker Graviton コンテナリストおよびリンクされた Deep Learning Containers の commit と比較し、古くなった PyTorch バージョンのエントリを更新します。完了条件は、SDK が Graviton デプロイ用に利用可能な 2.6.0 のイメージ URI を公開することです。

索引モデルが issue の本文から書いたものです。

評価

技術スタック
aws, python, pytorch
領域
cloud, machine-learning
issue の種類
バグ
難易度
2/5
見積もり時間
1〜3時間
活発さ
静か
明瞭さ
明確に書かれている
初心者へのやさしさ
74/100

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。