Merge pull request aws#433 from ChoiByungWook/fix_extend

Fix dockerfile to work in other regions

Author: djarpin

advanced_functionality/pytorch_extending_our_containers/container/Dockerfile

@@ -14,8 +14,10 @@
 # For more information on creating a Dockerfile
 # https://docs.docker.com/compose/gettingstarted/#step-2-create-a-dockerfile
 # https://github.com/awslabs/amazon-sagemaker-examples/master/advanced_functionality/pytorch_extending_our_containers/pytorch_extending_our_containers.ipynb
+ARG REGION=us-west-2
+
 # SageMaker PyTorch image
-FROM 520713654638.dkr.ecr.us-west-2.amazonaws.com/sagemaker-pytorch:0.4.0-cpu-py3
+FROM 520713654638.dkr.ecr.$REGION.amazonaws.com/sagemaker-pytorch:0.4.0-cpu-py3
 
 ENV PATH="/opt/ml/code:${PATH}"
 

advanced_functionality/pytorch_extending_our_containers/container/build_and_push.sh

@@ -47,7 +47,7 @@ $(aws ecr get-login --registry-ids 520713654638 --region ${region} --no-include-
 # Build the docker image locally with the image name and then push it to ECR
 # with the full name.
 
-docker build  -t ${image} .
+docker build  -t ${image} . --build-arg REGION=${region}
 docker tag ${image} ${fullname}
 
 docker push ${fullname}

advanced_functionality/pytorch_extending_our_containers/pytorch_extending_our_containers.ipynb

@@ -322,7 +322,7 @@
     "# Build the docker image locally with the image name and then push it to ECR\n",
     "# with the full name.\n",
     "\n",
-    "docker build  -t ${algorithm_name} .\n",
+    "docker build  -t ${algorithm_name} . --build-arg REGION=${region}\n",
     "docker tag ${algorithm_name} ${fullname}\n",
     "\n",
     "docker push ${fullname}"