
Machine Learning with Amazon SageMaker Cookbook
By :

In the previous recipe, we prepared and built the custom container image using the docker build
command. In this recipe, we will push the custom container image to an Amazon ECR repository. If this is your first time hearing about Amazon ECR, it is simply a fully managed container registry that helps us manage our container images. After pushing the container image to an Amazon ECR repository, we will use this image for training and deployment in the Using the custom R algorithm container image for training and inference with Amazon SageMaker Local Mode recipe.
Here are the prerequisites for this recipe:
The initial steps in this recipe focus on creating the ECR repository. Let's get started:
Figure 2.99 – Navigating to the ECR console
As we can see, we can use the search bar to quickly navigate to the Elastic Container Registry service.
Figure 2.100 – Create repository button
Here, the Create repository button is at the top right of the screen.
$IMAGE_NAME
from the Building and testing the custom R algorithm container image recipe. In this case, we will use chap02_r
:Figure 2.101 – Create repository form
Here, we have the Create repository form. For Visibility settings, we chose Private and we set the Tag immutability configuration to Disabled.
Figure 2.102 – Create repository button
Finally, to complete the repository creation process, click the Create repository button at the bottom of the page.
Figure 2.103 – Link to the ECR repository page
Here, we have a link under the Repository name column. Clicking this link should redirect us to a page containing details about the repository.
Figure 2.104 – View push commands button (upper right)
The View push commands button can be found at the top right of the page.
aws ecr get-login-password …
, from the dialog box:Figure 2.105 – Push commands dialog box
Here, we can see multiple commands that we can use. We will only need the first one (aws ecr get-login-password …
). Click the icon with two overlapping boxes at the right-hand side of the code box to copy the entire line to the clipboard.
Figure 2.106 – New Terminal
The preceding screenshot shows us how to create a new Terminal. We click the green plus button and then select New Terminal from the list of options. Note that the green plus button is right under the Editor pane.
ml-r
directory:cd /home/ubuntu/environment/opt/ml-r
ACCOUNT_ID=$(aws sts get-caller-identity | jq -r ".Account") echo $ACCOUNT_ID
IMAGE_URI
value and use the ECR repository name we specified while creating the repository in this recipe. In this case, we will run IMAGE_URI="chap02_r"
:IMAGE_URI="<insert ECR Repository URI>" TAG="1"
aws ecr get-login-password --region us-east-1 | docker login --username AWS --password-stdin $ACCOUNT_ID.dkr.ecr.us-east-1.amazonaws.com
Important note
Note that we have assumed that our repository is in the us-east-1
region. Feel free to modify this region in the command if needed. This applies to all the commands in this chapter.
docker tag
command:docker tag $IMAGE_URI:$TAG $ACCOUNT_ID.dkr.ecr.us-east-1.amazonaws.com/$IMAGE_URI:$TAG
docker push
command:docker push $ACCOUNT_ID.dkr.ecr.us-east-1.amazonaws.com/$IMAGE_URI:$TAG
Now that we have completed this recipe, we can proceed with using this custom algorithm container image with SageMaker in the next recipe. But before that, let's see how this works!
In the Building and testing the custom R algorithm container image recipe, we used docker build
to prepare the custom container image. In this recipe, we created an Amazon ECR repository and pushed our custom container image to it. We also used the docker push
command to push the custom container image we built to the ECR repository.
Important note
Don't forget to include the api.r
file inside the container when writing this Dockerfile
and running the build step. The Python counterpart recipe copies the train
and serve
scripts to the /opt/ml
directory inside the container, while the R recipe copies the train
, serve
, and api.r
files to the /opt/ml
directory. If the api.r
file is not included, the following line in the serve
script file will trigger an error and cause the script to fail: pr <- plumb("/opt/ml/api.r")
.