Skip to content

Commit f821f95

Browse files
authored
Update paths in build_containers.py (aws#254)
Update paths in build_containers.py
1 parent 93655ee commit f821f95

File tree

3 files changed

+7
-2
lines changed

3 files changed

+7
-2
lines changed

.gitignore

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,12 @@ cm.json
116116
cm.ts
117117
collections.json
118118

119+
# Wheel and tar files from running `python bin/build_containers.py`
120+
*.whl
121+
bin/sagemaker-containers/**/*.tar.gz
122+
119123
# outdirs
124+
data/
120125
default/
121126
newlogsRunTest*/
122127
ts_outputs*/

bin/sagemaker-containers/tensorflow/1.13.1/build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ aws s3 sync $TORNASOLE_BINARY_PATH tornasole-binary
3232
cp tornasole-binary/*.whl .
3333
export TORNASOLE_BINARY=`ls tornasole-*.whl`
3434

35-
export TF_ESTIMATOR_BINARY_LOCATION=https://tensorflow-aws.s3-us-west-2.amazonaws.com/1.13/Ubuntu/estimator/tensorflow_estimator-1.13.1-py2.py3-none-any.whl
35+
export TF_ESTIMATOR_BINARY_LOCATION=https://tensorflow-aws.s3-us-west-2.amazonaws.com/1.13/Ubuntu/estimator/tensorflow_estimator-1.13.0-py2.py3-none-any.whl
3636
curl -O $TF_ESTIMATOR_BINARY_LOCATION
3737

3838
build() {

bin/sagemaker-containers/tensorflow/1.14.0/build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ build() {
4242

4343
export ECR_REPO_NAME=tornasole-preprod-tf-1.14.0-$mode
4444
# this is aws tf
45-
export TF_BINARY_LOCATION=https://tensorflow-aws.s3-us-west-2.amazonaws.com/1.14/AmazonLinux/$mode/latest-patch-latest-patch/tensorflow-1.14.0-cp36-cp36m-linux_x86_64.whl
45+
export TF_BINARY_LOCATION=https://tensorflow-aws.s3-us-west-2.amazonaws.com/1.14/AmazonLinux/$mode/final/tensorflow-1.14.0-cp36-cp36m-linux_x86_64.whl
4646
rm -rf tf-$mode.whl
4747
curl -o tf-$mode.whl $TF_BINARY_LOCATION
4848
docker build -t $ECR_REPO_NAME:$ECR_TAG_NAME --build-arg py_version=3 \

0 commit comments

Comments
 (0)