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