How to docker pull AWS SageMaker public images

pulling-image.sh

#!/usr/bin/env bash

# All the images are available under the AWS account id bellow
ACCOUNT_ID=520713654638

TAG=1.7.0-gpu-py2

# get aws region
REGION=$(aws configure get region)

# Allows docker to access the repository via Docker login
$(aws ecr get-login --no-include-email --registry-ids ${SAGEMAKER_ACCOUNT_ID})

# Pulling the image
docker pull ${ACCOUNT_ID}.dkr.ecr.${REGION}.amazonaws.com/sagemaker-tensorflow:${TAG}