aws / aws/sagemaker-python-sdk
PyTorch image_uris for graviton is outdated
- 主要言語
- 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