Chemins de registre Docker et exemple de code - Amazon SageMaker

Chemins de registre Docker et exemple de code

Les rubriques suivantes répertorient le chemin de registre Docker et d'autres paramètres pour chacun des algorithmes fournis par Amazon SageMaker et les conteneurs DLC (Deep Learning Containers).

Utilisez le chemin comme suit :

  • Pour créer une tâche d'entraînement (create_training_job), spécifiez le chemin de registre Docker (TrainingImage) et le mode d'entrée d'entraînement (TrainingInputMode) pour l'image d'entraînement. Vous créez une tâche d'entraînement pour entraîner un modèle à l'aide d'un ensemble de données spécifique.

  • Pour créer un modèle (create_model), spécifiez le chemin de l'image d'inférence (PrimaryContainer Image) dans le registre Docker (Image). SageMaker lance les instances de calcul de machine learning qui sont basées sur la configuration de point de terminaison et déploie le modèle, qui comprend les artefacts (le résultat d'entraînement du modèle).

Note

Pour le chemin de registre, utilisez la balise de version :1 afin de vous assurer d'utiliser une version stable de l'algorithme/du conteneur DLC. Vous pouvez héberger de façon fiable un modèle entraîné à l'aide d'une image avec la balise :1 sur une image d'inférence possédant la balise :1. Le fait d'utiliser la balise :latest dans le chemin de registre vous permet de bénéficier de la version la plus récente de l'algorithme/du conteneur DLC, mais peut entraîner des problèmes de rétrocompatibilité. Évitez d'utiliser la balise :latest à des fins de production.

Important

Lorsque vous récupérez l'URI d'image XGBoost dans SageMaker, n'utilisez pas :latest ni :1 pour la balise d'URI de l'image. Vous devez spécifier l'une des Versions prises en charge pour choisir le conteneur XGBoost géré par SageMaker avec la version native du package XGBoost que vous souhaitez utiliser. Pour trouver la version du package migrée dans les conteneurs XGBoost de SageMaker, consultez Chemins de registre Docker et exemple de code, choisissez votre Région AWS et accédez à la section XGBoost (algorithme).

Pour rechercher le chemin d'accès du registre, choisissez la région AWS, puis l'algorithme ou le conteneur DLC.