Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 1 addition & 2 deletions docker/1.15.2/py3/Dockerfile.cpu
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,6 @@ ENV SAGEMAKER_TRAINING_MODULE=sagemaker_tensorflow_container.training:main
# Define framework-related package sources
ARG FRAMEWORK_SUPPORT_INSTALLABLE=sagemaker_tensorflow_training*.tar.gz
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
ARG SMDEBUG_URL=https://s3.amazonaws.com/kannanva-smdebug-0.7/smdebug-0.7.0b20200309-py2.py3-none-any.whl

RUN apt-get update \
&& apt-get install -y --no-install-recommends \
Expand Down Expand Up @@ -100,7 +99,7 @@ RUN pip install --no-cache-dir -U \
keras_preprocessing==1.1.0 \
keras==2.3.1 \
requests==2.22.0 \
${SMDEBUG_URL} \
smdebug==0.5.0.post0 \
sagemaker==1.50.17 \
sagemaker-experiments==0.1.3 \
mpi4py==3.0.2 \
Expand Down
3 changes: 2 additions & 1 deletion docker/1.15.2/py3/Dockerfile.gpu
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ ARG FRAMEWORK_SUPPORT_INSTALLABLE=sagemaker_tensorflow_training*.tar.gz
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
ARG SMDEBUG_URL=https://s3.amazonaws.com/kannanva-smdebug-0.7/smdebug-0.7.0b20200309-py2.py3-none-any.whl


RUN apt-get update \
&& apt-get install -y --no-install-recommends --allow-unauthenticated \
python3-dev \
Expand Down Expand Up @@ -137,7 +138,7 @@ RUN pip install --no-cache-dir -U \
keras_preprocessing==1.1.0 \
requests==2.22.0 \
keras==2.3.1 \
${SMDEBUG_URL} \
smdebug==0.5.0.post0 \
sagemaker==1.50.17 \
sagemaker-experiments==0.1.3 \
mpi4py==3.0.2 \
Expand Down