temporary fix of Horovod docker image (#73)

kubeflow · Jan 9, 2019 · d8283d5 · d8283d5
1 parent 80c541b
commit d8283d5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/examples/tensorflow-benchmarks/Dockerfile b/examples/tensorflow-benchmarks/Dockerfile
@@ -1,5 +1,9 @@
 FROM uber/horovod:0.15.2-tf1.12.0-torch1.0.0-py2.7
 
+# Temporary fix until Horovod pushes out a new release.
+# See https://github.com/uber/horovod/pull/700
+RUN sed -i '/^NCCL_SOCKET_IFNAME.*/d' /etc/nccl.conf
+
 RUN mkdir /tensorflow
 WORKDIR "/tensorflow"
 RUN git clone -b cnn_tf_v1.12_compatible https://github.com/tensorflow/benchmarks