move tfd at tensorflow 2.1

Author: hephaex

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

@@ -66,7 +66,7 @@ RUN python3 -m pip install --no-cache-dir --extra-index-url \
 RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
     HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL \
     HOROVOD_WITH_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITHOUT_MXNET=1 \
-    pip install --no-cache-dir horovod==0.19.4 && \
+    pip install --no-cache-dir horovod==0.19.5 && \
     ldconfig
 
 RUN python3 -m pip install --no-cache-dir \
@@ -78,9 +78,6 @@ RUN python3 -m pip install --no-cache-dir \
 Run python3 -m ipykernel install --display-name "TensorFlow 2.1 on Python 3.6 (CUDA 10.1)" && \
     cat /usr/local/share/jupyter/kernels/python3/kernel.json
 
-RUN jupyter nbextension enable --py --sys-prefix widgetsnbextension && \
-    jupyter nbextension enable --py tensorflow_model_analysis
-
 # Backend.AI specifics
 LABEL ai.backend.kernelspec="1" \
       ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \