diff --git a/examples/tensorflow-benchmarks/Dockerfile b/examples/tensorflow-benchmarks/Dockerfile index 1d2b229..b7d8a38 100644 --- a/examples/tensorflow-benchmarks/Dockerfile +++ b/examples/tensorflow-benchmarks/Dockerfile @@ -1,8 +1,8 @@ -FROM uber/horovod:0.13.11-tf1.10.0-torch0.4.0-py2.7 +FROM uber/horovod:0.15.2-tf1.12.0-torch1.0.0-py2.7 RUN mkdir /tensorflow WORKDIR "/tensorflow" -RUN git clone https://github.com/tensorflow/benchmarks +RUN git clone -b cnn_tf_v1.12_compatible https://github.com/tensorflow/benchmarks WORKDIR "/tensorflow/benchmarks" CMD mpirun \