fixed cudnn miss matching at TensorFlow 2.1

Author: hephaex

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

@@ -1,3 +1,4 @@
+
 FROM lablup/common-base:20.03-py36-cuda10.1
 
 # Install TensorFlow, Keras, PyTorch and MXNet
@@ -90,5 +91,18 @@ RUN jupyter nbextensions_configurator enable && \
     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" \
+      ai.backend.features="batch query uid-match user-input" \
+      ai.backend.base-distro="ubuntu16.04" \
+      ai.backend.resource.min.cpu="1" \
+      ai.backend.resource.min.mem="1g" \
+      ai.backend.resource.min.cuda.device=1 \
+      ai.backend.resource.min.cuda.shares=0.1 \
+      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"
+
 WORKDIR /home/work
 # vim: ft=dockerfile