aws / aws/sagemaker-python-sdk

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

Đang mở
#4,035 0 bình luận 0 reaction 1 người được giao Được @mollyheamazon nhận Xem trên GitHub
component: neo Pending information
Ngôn ngữ chính
Python
Star
2.3k
Fork
1.3k
Merge trung bình
1 ngày 22 giờ
Pull request đã merge (30 ngày)
35

Mô tả

**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

Hướng dẫn đóng góp

Mở hướng dẫn đóng góp

Đánh giá

Issue này chưa được đánh giá.

Nhận issue mới trong hộp thư của bạn

Bản tóm tắt ngắn những issue GitHub phù hợp với người mới.