add OpenCV

Author: hephaex

vendor/ngc-pytorch/Dockerfile.22.09-py3

@@ -12,56 +12,150 @@ ENV DEBIAN_FRONTEND=noninteractive \
     C_INCLUDE_PATH=/usr/include/gdal \
     LANG=C.UTF-8
 
-RUN apt-get update && \
+RUN apt-key adv --refresh-keys --keyserver keyserver.ubuntu.com && \
+    apt-get update && \
     apt-get install -y --no-install-recommends \
 	automake \
+	bison \
+	build-essential \
 	bzip2 \
+	ca-certificates \
 	cabextract \
+	chrpath \
+	devscripts \
+	dh-make \
+	dirmngr \
+	dkms \
+	dpatch \
+	ethtool \
+	fakeroot \
 	ffmpeg \
+	flex \
 	fonts-nanum \
 	fonts-nanum-coding \
 	fonts-nanum-extra \
+	g++ \
+	gcc \
+	gdal-bin \
 	gfortran \
+	gir1.2-gtk-3.0 \
+	git \
+	graphviz \
 	htop \
+	iputils-ping \
 	libasound2-dev \
 	libatlas-base-dev \
+	libavcodec-dev \
+	libavformat-dev \
 	libavresample-dev \
+	libboost-dev \
+	libboost-filesystem-dev \
+	libboost-system-dev \
+	libcairo-dev \
+	libcairo2-dev \
+	libcurl3-dev \
 	libdc1394-22-dev \
 	libeigen3-dev \
 	libfaac-dev \
+	libfftw3-dev \
+	libfreetype6-dev \
+	libfuse2 \
 	libgdal-dev \
+	libgeos++-dev \
+	libgeos-dev \
 	libgflags-dev \
+	libgfortran4 \
+	libgirepository1.0-dev \
+	libglib2.0-0 \
 	libgoogle-glog-dev \
 	libgphoto2-dev \
 	libgstreamer-plugins-base1.0-dev \
 	libgstreamer1.0-dev \
 	libgtk-3-dev \
-	libhdf5-dev \
+	libhdf5-dev \	
+	libhdf5-serial-dev \
+#	libjasper-dev \
 	libjemalloc-dev \
+	libjpeg-dev \
+	liblapack-dev \
+	libmnl0 \
 	libmp3lame-dev \
+	libmpdec2 \
 	libncurses-dev \
+	libnl-3-200 \
+	libnl-3-dev \
+	libnl-route-3-200 \
+	libnl-route-3-dev \
+	libnlopt-dev \
+	libnuma-dev \
 	libopenblas-dev \
 	libopencore-amrnb-dev \
 	libopencore-amrwb-dev \
+	libpci3 \
+	libpng-dev \
+	libportmidi-dev \
+	libproj-dev \
 	libprotobuf-dev \
+	libsdl-dev \
+	libsdl-image1.2-dev \
+	libsdl-mixer1.2-dev \
+	libsdl-ttf2.0-dev \
+	libsm6 \
+	libsmpeg-dev \
+	libssl-dev \
+	libswscale-dev \
+	libtbb-dev \
 	libtheora-dev \
+	libtiff-dev \
+	libv4l-dev \
 	libvorbis-dev \
 	libx264-dev \
 	libxext6 \
+	libxine2-dev \
+	libxml2-dev \
 	libxrender-dev \
+	libxrender1 \
+	libxslt1-dev \
 	libxvidcore-dev \
-	libsm6 \
-	libtbb-dev \
+	libzmq3-dev \
+	lintian \
+	lsof \
+	make \
 	mercurial \
+	mime-support \
+	mpich \
 	ncurses-term \
-        openjdk-8-jdk \
+	net-tools \
+	nodejs \
+	openjdk-8-jdk \
+	openssh-client \
+	openssh-server \
 	pbzip2 \
-	pv \
+	pciutils \
 	pdsh \
+	pkg-config \
+	proj-bin \
 	protobuf-compiler \
+	pv \
+	python-apt \
+	python-xkit \
+	python3-gdal \
+	screen-resolution-extra \
+	subversion \
+	swig \
+	tk \
+	unzip \
 	v4l-utils \	
-	x264 
-
+	vim \
+	x264 \
+	xvfb \
+	xz-utils \
+	yarn \
+	yasm \
+	zip \
+        tcl \
+        udev \
+	zlib1g-dev 
 # Install CUDA + cuDNN 
 RUN ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.8.3.2 /usr/local/cuda/lib64/libcudnn.so && \
     rm -rf /var/lib/apt/lists/* && \
@@ -112,9 +206,6 @@ RUN git clone -q --branch=v0.3.20 https://github.com/xianyi/OpenBLAS.git && \
     cd /tmp/scs-python && \
     python setup.py install --scs --gpu
 
-RUN /opt/conda/bin/conda install -n base conda==4.13.0
-RUN /opt/conda/bin/conda install -c conda-forge opencv ffmpeg 
-
 # install git-lfs
 WORKDIR /tmp
 RUN curl -sLO https://github.com/git-lfs/git-lfs/releases/download/v3.3.0/git-lfs-linux-amd64-v3.3.0.tar.gz && \
@@ -131,20 +222,68 @@ RUN git clone https://github.com/aristocratos/bashtop.git && \
 RUN curl -fL https://github.com/cdr/code-server/releases/download/v4.9.1/code-server-4.9.1-linux-amd64.tar.gz \
   | tar -C /usr/local/lib -xz && \ 
     mv /usr/local/lib/code-server-4.9.1-linux-amd64 /usr/local/lib/code-server-4.9.1 && \
-    ln -s /usr/local/lib/code-server-4.9.1/bin/code-server /usr/local/bin/code-server
+    ln -s /usr/local/lib/code-server-4.9.1/bin/code-server /usr/local/bin/code-server && \
+    code-server \
+        --install-extension ms-python.python \
+        --install-extension ms-ceintl.vscode-language-pack-ko
 
+# OpenCV
+ENV OPENCV_VERSION="4.5.5"
 WORKDIR /tmp
-COPY ./requirements.22.09.1.txt /tmp/requirements.txt
-RUN /opt/conda/bin/python3 -m pip install --no-cache-dir -r requirements.txt
-COPY ./requirements.22.09.2.txt /tmp/requirements.txt
-RUN /opt/conda/bin/python3 -m pip install --no-cache-dir -r requirements.txt
-COPY ./requirements.22.09.3.txt /tmp/requirements.txt
-RUN /opt/conda/bin/python3 -m pip install --no-cache-dir -r requirements.txt
-COPY ./requirements.22.09.4.txt /tmp/requirements.txt
-RUN /opt/conda/bin/python3 -m pip install --no-cache-dir -r requirements.txt
-COPY ./requirements.22.09.5.txt /tmp/requirements.txt
-RUN /opt/conda/bin/python3 -m pip install --no-cache-dir -r requirements.txt && \
-    rm -f /tmp/*.whl /tmp/requirem*
+RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
+    wget -O opencv-contrib.zip https://github.com/opencv/opencv_contrib/archive/${OPENCV_VERSION}.zip && \
+    unzip ${OPENCV_VERSION}.zip && \
+    unzip opencv-contrib.zip && \
+    mkdir opencv-${OPENCV_VERSION}/cmake_binary && \
+    cd opencv-${OPENCV_VERSION}/cmake_binary && \
+    cmake \
+      -DCMAKE_BUILD_TYPE=RELEASE \
+      -D BUILD_TIFF=ON \
+      -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.8 \
+      -D ENABLE_FAST_MATH=1 \
+      -D CUDA_FAST_MATH=1 \
+      -D WITH_CUBLAS=1 \
+      -D WITH_OPENGL=ON \
+      -D WITH_OPENCL=ON \
+      -D WITH_IPP=ON \
+      -D WITH_TBB=ON \
+      -D WITH_EIGEN=ON \
+      -D WITH_V4L=ON \
+      -D BUILD_TESTS=OFF \
+      -D BUILD_PERF_TESTS=OFF \
+      -D OPENCV_EXTRA_MODULES_PATH="../../opencv_contrib-4.5.5/modules" \
+      -D CMAKE_BUILD_TYPE=RELEASE \
+      -D CMAKE_INSTALL_PREFIX=$(python3 -c "import sys; print(sys.prefix)") \
+      -D PYTHON_EXECUTABLE=$(which python3) \
+      -D PYTHON_INCLUDE_DIR=$(python3 -c "from distutils.sysconfig import get_python_inc; print(get_python_inc())") \
+      -D PYTHON_PACKAGES_PATH=$(python3 -c "from distutils.sysconfig import get_python_lib; print(get_python_lib())") \
+      .. 2>&1 | tee cmake_messages.txt && \
+    make -j$(nproc) && \
+    make install && \
+    cd /tmp && \
+    python3 -m pip install --no-cache-dir opencv-python && \
+    rm -fr opencv* 
+
+WORKDIR /tmp
+COPY requirements.22.09.txt /tmp/requirements.txt
+RUN for f in $(ls requirements*.txt); do pip3 install --disable-pip-version-check --no-cache-dir -r $f; done && \
+     rm -f /tmp/*.whl /tmp/requirem*
+
+
+#COPY ./requirements.22.09.1.txt /tmp/requirements.txt
+#RUN /opt/conda/bin/python3 -m pip install --no-cache-dir -r requirements.txt
+#COPY ./requirements.22.09.2.txt /tmp/requirements.txt
+#RUN /opt/conda/bin/python3 -m pip install --no-cache-dir -r requirements.txt
+#COPY ./requirements.22.09.3.txt /tmp/requirements.txt
+#RUN /opt/conda/bin/python3 -m pip install --no-cache-dir -r requirements.txt
+#COPY ./requirements.22.09.4.txt /tmp/requirements.txt
+#RUN /opt/conda/bin/python3 -m pip install --no-cache-dir -r requirements.txt
+#COPY ./requirements.22.09.5.txt /tmp/requirements.txt
+#RUN /opt/conda/bin/python3 -m pip install --no-cache-dir -r requirements.txt && \
+#    rm -f /tmp/*.whl /tmp/requirem*
 
 # Install Open MPI
 RUN mkdir /tmp/openmpi && \
@@ -179,6 +318,16 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
     pip install --no-cache-dir horovod==0.24.2 && \
     ldconfig
 
+RUN python3 -m pip install pytorch-lightning && \
+    python3 -m pip install --no-cache-dir \
+        pyg-lib \
+	torch-scatter \
+	torch-sparse \
+	torch-cluster \
+	torch-spline-conv \
+	torch-geometric \
+        -f https://data.pyg.org/whl/torch-1.13.0+cu117.html 
+
 RUN python3 -m pip install --no-cache-dir \
     	mpi4py==3.1.3 \
 	nni==2.9 \