update nni

Author: hephaex

vendor/ngc-pytorch/Dockerfile.21.03-py3

@@ -15,34 +15,46 @@ ENV DEBIAN_FRONTEND=noninteractive \
     LANG=C.UTF-8
 
 RUN apt-get update && \
-    apt-get install -y \
-        libsm6 libxext6 libxrender-dev mercurial libopenblas-dev libgdal-dev \
-       	openjdk-8-jdk \
-	libasound2-dev \
+    apt-get install -y --no-install-recommends \
+	automake \
+	fonts-nanum \
+	fonts-nanum-coding \
+	fonts-nanum-extra \
 	gfortran \
-	libgstreamer1.0-dev \
-	libgstreamer-plugins-base1.0-dev \
-	libgtk-3-dev \
-        libtbb-dev \
+	htop \
+	libasound2-dev \
 	libatlas-base-dev \
+	libavresample-dev \
 	libdc1394-22-dev \
-	libxvidcore-dev \
+	libeigen3-dev \
 	libfaac-dev \
+	libgdal-dev \
+	libgflags-dev \
+	libgoogle-glog-dev \
+	libgphoto2-dev \
+	libgstreamer-plugins-base1.0-dev \
+	libgstreamer1.0-dev \
+	libgtk-3-dev \
+	libhdf5-dev \	
 	libmp3lame-dev \
+	libopenblas-dev \
+	libopencore-amrnb-dev \
+	libopencore-amrwb-dev \
+	libprotobuf-dev \
 	libtheora-dev \
 	libvorbis-dev \
-	libxvidcore-dev \
-	libopencore-amrnb-dev libopencore-amrwb-dev \
-	libavresample-dev \
-	x264 \
 	libx264-dev \
+	libxext6 \
+	libxrender-dev \
+	libxvidcore-dev \
+	mercurial \
+	ncurses-term \
+	protobuf-compiler \
 	v4l-utils \	
-	libprotobuf-dev protobuf-compiler \
-	libgoogle-glog-dev libgflags-dev \
-	libgphoto2-dev \
-	libeigen3-dev \
-	libhdf5-dev \	
-	automake 
+	x264 \
+       	openjdk-8-jdk \
+        libsm6 \
+        libtbb-dev 
 
 
 RUN update-alternatives --install /opt/conda/bin/python python /opt/conda/bin/python3 2
@@ -88,7 +100,7 @@ RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/videodev.h && \
       -D BUILD_opencv_java=OFF \
       -D WITH_CUDA=ON \
       -D CUDA_NVCC_FLAGS=--expt-relaxed-constexpr \
-      -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-11.1 \
+      -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-11.2 \
       -D ENABLE_FAST_MATH=1 \
       -D CUDA_FAST_MATH=1 \
       -D WITH_CUBLAS=1 \
@@ -165,10 +177,15 @@ RUN curl -sLO https://github.com/git-lfs/git-lfs/releases/download/v2.13.3/git-l
     bash install.sh && \
     rm -rf /tmp/*
 
-COPY ./service-defs /etc/backend.ai/service-defs
+WORKDIR /tmp
+RUN git clone https://github.com/aristocratos/bashtop.git && \
+    cd bashtop && \
+    make install
+
 RUN curl -fL https://github.com/cdr/code-server/releases/download/v3.9.3/code-server-3.9.3-linux-amd64.tar.gz | tar -C /usr/local/lib -xz && \
     mv /usr/local/lib/code-server-3.9.3-linux-amd64 /usr/local/lib/code-server-3.9.3 && \
     ln -s /usr/local/lib/code-server-3.9.3/bin/code-server /usr/local/bin/code-server
+
 # Install Open MPI
 RUN mkdir /tmp/openmpi && \
     cd /tmp/openmpi && \
@@ -198,7 +215,7 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
 
 RUN python3 -m pip install --no-cache-dir \
     	    mpi4py==3.0.3 \
-	    nni==2.0 \
+	    nni==2.1 \
 	    mlflow==1.15.0 \
 	    scikit-nni==0.1.1
 
@@ -218,6 +235,7 @@ RUN jupyter nbextensions_configurator enable && \
 
 RUN apt autoclean && \
     sed -i 's/source \/usr\/local\/nvm\/nvm.sh//' /etc/bash.bashrc && \
+    ln -sf /usr/share/terminfo/x/xterm-color /usr/share/terminfo/x/xterm-256color && \
     rm -rf /var/lib/apt/lists/* && \	
     rm -rf /root/.cache && \
     rm -rf /tmp/*