opencv

Author: hephaex

commons/Dockerfile.base.py38-cuda11.3

@@ -307,7 +307,7 @@ RUN ln -s /usr/include/libv4l1-videodev.h /usr/include/linux/videodev.h && \
     rm -rf /var/lib/apt/lists/
 
 WORKDIR /tmp
-ENV OPENCV_VERSION="4.5.1"
+ENV OPENCV_VERSION="4.5.2"
 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 && \
@@ -318,11 +318,19 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
       -DCMAKE_BUILD_TYPE=RELEASE \
       -D BUILD_TIFF=ON \
       -D BUILD_opencv_java=OFF \
+      -D CUDA_BIN_PATH=/usr/local/cuda \
+      -D CUDA_CUDA_LIBRARY=/usr/lib/x86_64-linux-gnu/stubs/libcuda.so \
+      -D CUDA_FAST_MATH=ON \
+      -D CUDA_HOST_COMPILER=/usr/bin/gcc-6 \
+      -D CUDA_NVCC_FLAGS="-D_FORCE_INLINES --expt-relaxed-constexpr" \
+      -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-10.1 \
+      -D CUDNN_INCLUDE_DIR=/usr/local/cuda/include \
+      -D CUDNN_LIBRARY=/usr/lib/x86_64-linux-gnu/libcudnn.so.8.0.5.39 \
+      -D ENABLE_AVX=ON \
       -D WITH_CUDA=ON \
       -D CUDA_NVCC_FLAGS=--expt-relaxed-constexpr \
       -D CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-11.3 \
       -D ENABLE_FAST_MATH=1 \
-      -D CUDA_FAST_MATH=1 \
       -D WITH_CUBLAS=1 \
       -D WITH_OPENGL=ON \
       -D WITH_OPENCL=ON \
@@ -332,7 +340,7 @@ RUN wget https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip && \
       -D WITH_V4L=ON \
       -D BUILD_TESTS=OFF \
       -D BUILD_PERF_TESTS=OFF \
-      -D OPENCV_EXTRA_MODULES_PATH="../../opencv_contrib-4.5.1/modules" \
+      -D OPENCV_EXTRA_MODULES_PATH="../../opencv_contrib-4.5.2/modules" \
       -D CMAKE_BUILD_TYPE=RELEASE \
       -D CMAKE_INSTALL_PREFIX=$(python3 -c "import sys; print(sys.prefix)") \
       -D PYTHON_EXECUTABLE=$(which python3) \
@@ -426,14 +434,12 @@ RUN jupyter nbextensions_configurator enable && \
     jupyter labextension install --no-build @jupyterlab/toc && \
     jupyter nbextension enable execute_time/ExecuteTime && \
     jupyter nbextension enable toc2/main && \
-#    jupyter labextension install @pyviz/jupyterlab_pyviz && \
-    jupyter labextension install @bokeh/jupyter_bokeh && \
-    jupyter labextension install --no-build jupyterlab-nvdashboard && \
-    jupyter lab build
+    jupyter labextension install @jupyterlab/toc && \
+    jupyter lab build    
 
-# Clean up
 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/*