Update build script for PT container

- modified S3 path to pick up from PT folder
- added parameter to enable installation of sagemaker_pytorch_container.whl into image

Author: ddavydenko

bin/sagemaker-containers/pytorch/1.1.0/build.sh

@@ -23,20 +23,23 @@ popd
 cd bin/sagemaker-containers/pytorch/1.1.0/
 cp ~/sagemaker-pytorch-container/dist/sagemaker_pytorch_container-*.whl .
 cp -r ~/sagemaker-pytorch-container/lib .
+export SM_BINARY=`ls sagemaker_pytorch_container-*.whl`
 
-export TORNASOLE_BINARY_PATH=s3://tornasole-binaries-use1/tornasole_mxnet/py3/latest
+export TORNASOLE_BINARY_PATH=s3://tornasole-binaries-use1/tornasole_pytorch/py3/latest
 aws s3 sync $TORNASOLE_BINARY_PATH tornasole-binary
 cp tornasole-binary/*.whl .
 export TORNASOLE_BINARY=`ls tornasole-*.whl`
 
 export ECR_REPO_NAME=tornasole-preprod-pytorch-1.1.0-gpu
 docker build -t $ECR_REPO_NAME:$ECR_TAG_NAME --build-arg py_version=3 \
     --build-arg tornasole_framework_installable=$TORNASOLE_BINARY \
+    --build-arg framework_support_installable=$SM_BINARY \
     -f Dockerfile.gpu .
 tag_and_push
 
 export ECR_REPO_NAME=tornasole-preprod-pytorch-1.1.0-cpu
 docker build -t $ECR_REPO_NAME:$ECR_TAG_NAME --build-arg py_version=3 \
     --build-arg tornasole_framework_installable=$TORNASOLE_BINARY \
+    --build-arg framework_support_installable=$SM_BINARY \
     -f Dockerfile.cpu .
-tag_and_push
+tag_and_push