File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -24,6 +24,7 @@ ENV SAGEMAKER_TRAINING_MODULE=sagemaker_tensorflow_container.training:main
24
24
# Define framework-related package sources
25
25
ARG FRAMEWORK_SUPPORT_INSTALLABLE=sagemaker_tensorflow_training*.tar.gz
26
26
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
27
28
28
29
RUN apt-get update \
29
30
&& apt-get install -y --no-install-recommends \
@@ -99,7 +100,7 @@ RUN pip install --no-cache-dir -U \
99
100
keras_preprocessing==1.1.0 \
100
101
keras==2.3.1 \
101
102
requests==2.22.0 \
102
- smdebug==0.5.0.post0 \
103
+ ${SMDEBUG_URL} \
103
104
sagemaker==1.50.17 \
104
105
sagemaker-experiments==0.1.3 \
105
106
mpi4py==3.0.2 \
Original file line number Diff line number Diff line change @@ -21,6 +21,7 @@ ENV SAGEMAKER_TRAINING_MODULE=sagemaker_tensorflow_container.training:main
21
21
# Define framework-related package sources
22
22
ARG FRAMEWORK_SUPPORT_INSTALLABLE=sagemaker_tensorflow_training*.tar.gz
23
23
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
24
25
25
26
RUN apt-get update \
26
27
&& apt-get install -y --no-install-recommends --allow-unauthenticated \
@@ -136,7 +137,7 @@ RUN pip install --no-cache-dir -U \
136
137
keras_preprocessing==1.1.0 \
137
138
requests==2.22.0 \
138
139
keras==2.3.1 \
139
- smdebug==0.5.0.post0 \
140
+ ${SMDEBUG_URL} \
140
141
sagemaker==1.50.17 \
141
142
sagemaker-experiments==0.1.3 \
142
143
mpi4py==3.0.2 \
You can’t perform that action at this time.
0 commit comments