Skip to content

Commit 42762fe

Browse files
authored
Revert "Update smdebug to 0.7.0 - TF 1.15.2 (#298)" (#303)
This reverts commit 44656fc.
1 parent d451fbe commit 42762fe

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

docker/1.15.2/py3/Dockerfile.cpu

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@ 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.0b20200309-py2.py3-none-any.whl
2827

2928
RUN apt-get update \
3029
&& apt-get install -y --no-install-recommends \
@@ -100,7 +99,7 @@ RUN pip install --no-cache-dir -U \
10099
keras_preprocessing==1.1.0 \
101100
keras==2.3.1 \
102101
requests==2.22.0 \
103-
${SMDEBUG_URL} \
102+
smdebug==0.5.0.post0 \
104103
sagemaker==1.50.17 \
105104
sagemaker-experiments==0.1.3 \
106105
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
@@ -23,6 +23,7 @@ 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
2424
ARG SMDEBUG_URL=https://s3.amazonaws.com/kannanva-smdebug-0.7/smdebug-0.7.0b20200309-py2.py3-none-any.whl
2525

26+
2627
RUN apt-get update \
2728
&& apt-get install -y --no-install-recommends --allow-unauthenticated \
2829
python3-dev \
@@ -137,7 +138,7 @@ RUN pip install --no-cache-dir -U \
137138
keras_preprocessing==1.1.0 \
138139
requests==2.22.0 \
139140
keras==2.3.1 \
140-
${SMDEBUG_URL} \
141+
smdebug==0.5.0.post0 \
141142
sagemaker==1.50.17 \
142143
sagemaker-experiments==0.1.3 \
143144
mpi4py==3.0.2 \

0 commit comments

Comments
 (0)