horovod 0.21.3

Author: hephaex

python-tensorflow/Dockerfile.2.3-py36-cuda10.1

@@ -2,6 +2,8 @@ FROM lablup/common-base:cuda10.1
 
 WORKDIR /tmp
 COPY ./requirements.tf.txt /tmp
+RUN python3 -m pip uninstall --no-cache-dir -y \
+        tensorflow tensorflow-serving tensorflow-text
 RUN python3 -m pip install --no-cache-dir \
         -r  requirements.tf.txt \
 	--ignore-installed && \
@@ -18,7 +20,7 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
     HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL HOROVOD_NCCL_LINK=SHARED \
     HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1\
     HOROVOD_GPU=CUDA \
-    python3 -m pip install --no-cache-dir horovod==0.21.1 && \
+    python3 -m pip install --no-cache-dir horovod==0.21.3 && \
     ldconfig