aws / aws/sagemaker-python-sdk

Request image_uris.retrieve utility to include inf2 / neuron containers.

オープン
#4,035 コメント 0 件 リアクション 0 件 担当者 1 名 @mollyheamazon が担当を希望しています GitHub で見る
component: neo Pending information
主要言語
Python
スター
2.3k
フォーク
1.3k
平均マージ
1日 22時間
マージ済み PR(30日)
35

説明

**Describe the feature you'd like**
Request image_uris.retrieve utility to include inf2 containers.
```
baseimage = sagemaker.image_uris.retrieve(
framework="Neuron",
region=region,
py_version="py310",
image_scope="inference",
version="2.12.0",
instance_type="ml.inf2.24xlarge",
)
print(baseimage)

FileNotFoundError: [Errno 2] No such file or directory: '/home/ec2-user/anaconda3/envs/python3/lib/python3.10/site-packages/sagemaker/image_uri_config/Neuron.json'
```
Need to update this: src/sagemaker/image_uri_config/

Output should be the correct ECR image:

eg. 763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference-neuronx:1.13.1-neuronx-py310-sdk2.12.0-ubuntu20.04

**Describe alternatives you've considered**
Manually searching in https://github.com/aws/deep-learning-containers/blob/master/available_images.md#neuron-containers

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

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

評価

この issue はまだ評価されていません。

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

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