From 9605bef7c1bdd64bd77009551d4d6b9cc1e18b76 Mon Sep 17 00:00:00 2001 From: Alex Sergeev Date: Mon, 19 Aug 2019 21:02:01 -0700 Subject: [PATCH] Update Fashion MNIST tutorial to Horovod 0.17.0 Signed-off-by: Alex Sergeev --- fashion_mnist/Dockerfile | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/fashion_mnist/Dockerfile b/fashion_mnist/Dockerfile index 71c421b..13a8dea 100644 --- a/fashion_mnist/Dockerfile +++ b/fashion_mnist/Dockerfile @@ -1,4 +1,4 @@ -FROM horovod/horovod:0.16.0-tf1.12.0-torch1.0.0-mxnet1.4.0-py3.5 +FROM horovod/horovod:0.17.0.post1-tf1.14.0-torch1.2.0-mxnet1.5.0-py3.6 RUN pip install jupyterlab @@ -8,9 +8,6 @@ RUN ldconfig /usr/local/cuda/lib64/stubs && \ python -c 'import keras; keras.datasets.fashion_mnist.load_data()' && \ ldconfig -# Until https://github.com/horovod/horovod/pull/858 is merged -RUN ln -s /usr/local/bin/mpirun /usr/local/bin/horovodrun - COPY . /tutorial WORKDIR /tutorial