Skip to content

Commit 8672431

Browse files
icywang86ruiEliza Zhang
authored andcommitted
Upgrade to TensorFlow 1.13.1 (aws#184)
* Upgrade to TensorFlow 1.13.1 * Add 1.13.1 dockerfiles * Lower process number for one of the horovod local mode test Too many processes running in parallel causes the test fail sometimes. * Use multiprocessing.Process to start ps
1 parent 7cea030 commit 8672431

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

docker/1.13.1/Dockerfile.cpu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,4 +107,4 @@ RUN ${PIP} install --no-cache-dir -U \
107107

108108
ENV SAGEMAKER_TRAINING_MODULE sagemaker_tensorflow_container.training:main
109109

110-
CMD ["bin/bash"]
110+
CMD ["bin/bash"]

src/sagemaker_tensorflow_container/training.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
import multiprocessing
1818
import os
1919
import subprocess
20-
import threading
2120
import time
2221

2322
import sagemaker_containers.beta.framework as framework

0 commit comments

Comments
 (0)