upgrade tensorflow 2.1.1

Author: hephaex

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

@@ -2,14 +2,7 @@
 FROM lablup/common-base:20.03-py36-cuda10.1
 
 # Install TensorFlow, Keras, PyTorch and MXNet
-ENV TENSORFLOW_VERSION=2.1.0
-#ARG CUDA=10.1
-#ARG LIBNVINFER=6.0.1-1
-#ARG LIBNVINFER_MAJOR_VERSION=6
-#ENV CUDNN_VERSION=7.6.4.38-1+cuda10.1
-#ENV NCCL_VERSION=2.4.8-1+cuda10.1
-ENV DEBIAN_FRONTEND=noninteractive
-ENV mecab_dicdir /usr/local/lib/mecab/dic/mecab-ko-dic
+ENV TENSORFLOW_VERSION=2.1.1
 
 ENV PYTHONUNBUFFERED=1 \
     LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/lib64:/usr/local/nvidia/lib64:/usr/include/x86_64-linux-gnu" \
@@ -77,20 +70,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
 
-# Jupyter notebook extension
-RUN mkdir -p /home/work/.jupyter/nbextension
-WORKDIR /home/work/.jupyter/nbextension
-
-RUN jupyter nbextensions_configurator enable && \
-    jupyter contrib nbextension install && \
-    jupyter nbextension enable --py --sys-prefix widgetsnbextension && \
-    jupyter contrib nbextension install && \
-    jupyter serverextension enable --py jupyterlab --sys-prefix && \
-    jupyter labextension install @jupyter-widgets/jupyterlab-manager && \
-#    jupyter labextension install jupyterlab-nvdashboard && \
-    git clone https://github.com/lambdalisue/jupyter-vim-binding vim_binding && \
-    jupyter nbextension enable /home/work/.jupyter/nbextension/vim_binding/vim_binding 
-
 # Backend.AI specifics
 LABEL ai.backend.kernelspec="1" \
       ai.backend.envs.corecount="OPENBLAS_NUM_THREADS,OMP_NUM_THREADS,NPROC" \
@@ -102,7 +81,7 @@ LABEL ai.backend.kernelspec="1" \
       ai.backend.resource.min.cuda.shares=0 \
       ai.backend.runtime-type="python" \
       ai.backend.runtime-path="/usr/bin/python3" \
-      ai.backend.service-ports="ipython:pty:3000,tensorboard:http:6006,jupyter:http:8080,jupyterlab:http:8090"
+      ai.backend.service-ports="jupyter:http:8080,jupyterlab:http:8090,ipython:pty:3000,tensorboard:http:6006"
 
 WORKDIR /home/work
 # vim: ft=dockerfile