Images de conteneur d'inférence - Amazon SageMaker

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

Images de conteneur d'inférence

SageMaker Neo fournit désormais les informations d'URI des images d'inférence pour les cibles ml_*. Pour plus d'informations, consultez DescribeCompilationJob.

Selon votre cas d'utilisation, remplacez la partie en surbrillance dans le modèle d'URI d'image d'inférence fourni ci-dessous par les valeurs qui conviennent.

aws_account_id.dkr.ecr.aws_region.amazonaws.com/xgboost-neo:latest

Remplacez aws_account_id à partir du tableau en fin de page, selon la aws_region que vous avez utilisée.

aws_account_id.dkr.ecr.aws_region.amazonaws.com/sagemaker-neo-keras:fx_version-instance_type-py3

Remplacez aws_account_id à partir du tableau en fin de page, selon la aws_region que vous avez utilisée.

Remplacez fx_version par 2.2.4.

Remplacez instance_type par cpu ou gpu.

CPU or GPU instance types
aws_account_id.dkr.ecr.aws_region.amazonaws.com/sagemaker-inference-mxnet:fx_version-instance_type-py3

Remplacez aws_account_id à partir du tableau en fin de page, selon la aws_region que vous avez utilisée.

Remplacez fx_version par 1.8.0.

Remplacez instance_type par cpu ou gpu.

Inferentia1
aws_account_id.dkr.ecr.aws_region.amazonaws.com/sagemaker-neo-mxnet:fx_version-instance_type-py3

Remplacez aws_region par us-east-1 ou us-west-2.

Remplacez aws_account_id à partir du tableau en fin de page, selon la aws_region que vous avez utilisée.

Remplacez fx_version par 1.5.1.

Remplacez instance_type par inf.

aws_account_id.dkr.ecr.aws_region.amazonaws.com/sagemaker-neo-onnx:fx_version-instance_type-py3

Remplacez aws_account_id à partir du tableau en fin de page, selon la aws_region que vous avez utilisée.

Remplacez fx_version par 1.5.0.

Remplacez instance_type par cpu ou gpu.

CPU or GPU instance types
aws_account_id.dkr.ecr.aws_region.amazonaws.com/sagemaker-inference-pytorch:fx_version-instance_type-py3

Remplacez aws_account_id à partir du tableau en fin de page, selon la aws_region que vous avez utilisée.

Remplacez fx_version par 1.4, 1.5, 1.6, 1.7, 1.8, 1.12, 1.13 ou 2.0.

Remplacez instance_type par cpu ou gpu.

Inferentia1
aws_account_id.dkr.ecr.aws_region.amazonaws.com/sagemaker-neo-pytorch:fx_version-instance_type-py3

Remplacez aws_region par us-east-1 ou us-west-2.

Remplacez aws_account_id à partir du tableau en fin de page, selon la aws_region que vous avez utilisée.

Remplacez fx_version par 1.5.1.

Remplacez instance_type par inf.

Inferentia2 and Trainium1
763104351884.dkr.ecr.aws_region.amazonaws.com/pytorch-inference-neuronx:1.13.1-neuronx-py38-sdk2.10.0-ubuntu20.04

Remplacez aws_region par us-east-2 pour Inferentia2 et us-east-1 pour Trainium1.

CPU or GPU instance types
aws_account_id.dkr.ecr.aws_region.amazonaws.com/sagemaker-inference-tensorflow:fx_version-instance_type-py3

Remplacez aws_account_id à partir du tableau en fin de page, selon la aws_region que vous avez utilisée.

Remplacez fx_version par 1.15.3 ou 2.9.

Remplacez instance_type par cpu ou gpu.

Inferentia1
aws_account_id.dkr.ecr.aws_region.amazonaws.com/sagemaker-neo-tensorflow:fx_version-instance_type-py3

Remplacez aws_account_id à partir du tableau en fin de page, selon la aws_region que vous avez utilisée. Veuillez noter que, pour type d'instance inf, seuls us-east-1 et us-west-2 sont pris en charge.

Remplacez fx_version par 1.15.0

Remplacez instance_type par inf.

Inferentia2 and Trainium1
763104351884.dkr.ecr.aws_region.amazonaws.com/tensorflow-inference-neuronx:2.10.1-neuronx-py38-sdk2.10.0-ubuntu20.04

Remplacez aws_region par us-east-2 pour Inferentia2 et us-east-1 pour Trainium1.

Le tableau suivant mappe aws_account_id avec aws_region. Utilisez ce tableau pour trouver l'URI d'image d'inférence correcte dont vous avez besoin pour votre application.

aws_account_id aws_region
785573368785 us-east-1
007439368137 us-east-2
710691900526 us-west-1
301217895009 us-west-2
802834080501 eu-west-1
205493899709 eu-west-2
254080097072 eu-west-3
601324751636 eu-north-1
966458181534 eu-south-1
746233611703 eu-central-1
110948597952 ap-east-1
763008648453 ap-south-1
941853720454 ap-northeast-1
151534178276 ap-northeast-2
925152966179 ap-northeast-3
324986816169 ap-southeast-1
355873309152 ap-southeast-2
474822919863 cn-northwest-1
472730292857 cn-north-1
756306329178 sa-east-1
464438896020 ca-central-1
836785723513 me-south-1
774647643957 af-south-1
275950707576 il-central-1