Skip to content

Commit 44656fc

Browse files
authored
Update smdebug to 0.7.0 - TF 1.15.2 (#298)
1 parent bfc9c41 commit 44656fc

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

docker/1.15.2/py3/Dockerfile.cpu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ ENV SAGEMAKER_TRAINING_MODULE=sagemaker_tensorflow_container.training:main
2424
# Define framework-related package sources
2525
ARG FRAMEWORK_SUPPORT_INSTALLABLE=sagemaker_tensorflow_training*.tar.gz
2626
ARG TF_URL=https://tensorflow-aws.s3-us-west-2.amazonaws.com/1.15.2/AmazonLinux/cpu/final/tensorflow-1.15.2-cp36-cp36m-manylinux2010_x86_64.whl
27+
ARG SMDEBUG_URL=https://s3.amazonaws.com/kannanva-smdebug-0.7/smdebug-0.7.0b20200304-py2.py3-none-any.whl
2728

2829
RUN apt-get update \
2930
&& apt-get install -y --no-install-recommends \
@@ -99,7 +100,7 @@ RUN pip install --no-cache-dir -U \
99100
keras_preprocessing==1.1.0 \
100101
keras==2.3.1 \
101102
requests==2.22.0 \
102-
smdebug==0.5.0.post0 \
103+
${SMDEBUG_URL} \
103104
sagemaker==1.50.17 \
104105
sagemaker-experiments==0.1.3 \
105106
mpi4py==3.0.2 \

docker/1.15.2/py3/Dockerfile.gpu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ ENV SAGEMAKER_TRAINING_MODULE=sagemaker_tensorflow_container.training:main
2121
# Define framework-related package sources
2222
ARG FRAMEWORK_SUPPORT_INSTALLABLE=sagemaker_tensorflow_training*.tar.gz
2323
ARG TF_URL=https://tensorflow-aws.s3-us-west-2.amazonaws.com/1.15.2/AmazonLinux/gpu/final/tensorflow_gpu-1.15.2-cp36-cp36m-manylinux2010_x86_64.whl
24+
ARG SMDEBUG_URL=https://s3.amazonaws.com/kannanva-smdebug-0.7/smdebug-0.7.0b20200304-py2.py3-none-any.whl
2425

2526
RUN apt-get update \
2627
&& apt-get install -y --no-install-recommends --allow-unauthenticated \
@@ -136,7 +137,7 @@ RUN pip install --no-cache-dir -U \
136137
keras_preprocessing==1.1.0 \
137138
requests==2.22.0 \
138139
keras==2.3.1 \
139-
smdebug==0.5.0.post0 \
140+
${SMDEBUG_URL} \
140141
sagemaker==1.50.17 \
141142
sagemaker-experiments==0.1.3 \
142143
mpi4py==3.0.2 \

0 commit comments

Comments
 (0)