change base kernel 20.06 cuda10.1

Author: hephaex

python-mxnet/Dockerfile.1.6-py36-cuda10.1

@@ -1,9 +1,9 @@
-FROM lablup/common-base:20.03-py36-cuda10
+FROM lablup/common-base:20.06-py36-cuda10.1
 
 # Install TensorFlow
 ENV MXNET_VERSION=1.6.0
 
-RUN python3 -m pip install --no-cache-dir mxnet-cu100mkl==${MXNET_VERSION}
+RUN python3 -m pip install --no-cache-dir mxnet-cu101mkl==${MXNET_VERSION}
 RUN python3 -m pip install --no-cache-dir --extra-index-url \
     	    https://developer.download.nvidia.com/compute/redist/cuda/10.0 \
 	    nvidia-dali 	    
@@ -12,27 +12,18 @@ 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_WITHOUT_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITH_MXNET=1 \
-    pip install --no-cache-dir horovod==0.19.0 && \
+    pip install --no-cache-dir horovod==0.19.4 && \
     ldconfig
 
 RUN python3 -m pip install --no-cache-dir \
     	    mpi4py==3.0.3 \
-	    nni
-	    
+	    nni==1.6 \
+	    scikit-nni==0.2.1
+
 # Install ipython kernelspec
-RUN python3 -m ipykernel install --display-name "MXNet 1.6.0 on Python 3.6 (CUDA 10.0)" && \
+RUN python3 -m ipykernel install --display-name "MXNet 1.6.0 on Python 3.6 (CUDA 10.1 & MKL)" && \
     cat /usr/local/share/jupyter/kernels/python3/kernel.json
 
-# Install Horovod, temporarily using CUDA stubs
-RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
-    HOROVOD_GPU_ALLREDUCE=NCCL HOROVOD_GPU_BROADCAST=NCCL \
-    HOROVOD_WITHOUT_TENSORFLOW=1 HOROVOD_WITHOUT_PYTORCH=1 HOROVOD_WITH_MXNET=1 \
-    pip install --no-cache-dir horovod==0.19.0 && \
-    ldconfig
-
-RUN python3 -m pip install --no-cache-dir \
-    	    mpi4py==3.0.3 \
-	    nni
 
 # Backend.AI specifics
 LABEL ai.backend.kernelspec="1" \
@@ -45,7 +36,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="jupyter:http:8080,jupyterlab:http:8090,ipython:pty:3000,tensorboard:http:6006"
+      ai.backend.service-ports="jupyter:http:8080,jupyterlab:http:8090,ipython:pty:3000"
 
 WORKDIR /home/work
 # vim: ft=dockerfile