File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -24,7 +24,6 @@ ENV SAGEMAKER_TRAINING_MODULE=sagemaker_tensorflow_container.training:main
2424# Define framework-related package sources
2525ARG FRAMEWORK_SUPPORT_INSTALLABLE=sagemaker_tensorflow_training*.tar.gz
2626ARG 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
2928RUN 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 \
Original file line number Diff line number Diff line change @@ -23,6 +23,7 @@ ARG FRAMEWORK_SUPPORT_INSTALLABLE=sagemaker_tensorflow_training*.tar.gz
2323ARG 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
2424ARG SMDEBUG_URL=https://s3.amazonaws.com/kannanva-smdebug-0.7/smdebug-0.7.0b20200309-py2.py3-none-any.whl
2525
26+
2627RUN 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 \
You can’t perform that action at this time.
0 commit comments