Ubuntu16编译OpenCV4.4+CUDA10.1+Conda python3.8踩坑

说明

记录踩的坑，不详细记录过程。

安装

安装过程和设置主要参考了OpenCV官网教程。安装之前最好用apt安装一些必要的库，直接网上搜就能搜到。

环境

Ubuntu16 + CUDA10.1 + CUDNN8.0.4 + OpenCV4.4 (+contrib)
gcc 5.4, Conda python3.8

坑点

先放我的cmake配置命令：
cmake -DOPENCV_EXTRA_MODULES_PATH=/home/_/workspace/Env/sourcecode/opencv_contrib-4.4.0/modules -DBUILD_opencv_xfeatures2d=OFF -DBUILD_opencv_matlab=OFF \ -DBUILD_opencv_stereo=OFF -DBUILD_opencv_face=OFF -DCPU_BASELINE=SSE3,AVX2 -DCPU_DISPATCH=AVX,AVX2 -DWITH_CUDA=ON -DWITH_NVCUVID=ON \ -DCMAKE_INSTALL_PREFIX=/home/_/workspace/Env/opencv -DBUILD_opencv_python2=OFF -DBUILD_opencv_python3=ON \ -DPYTHON_DEFAULT_EXECUTABLE=/home/_/miniconda3/envs/newtorch/bin/python -DENABLE_PYLINT=ON \ -DPYTHON3_EXECUTABLE=/home/_/miniconda3/envs/newtorch/bin/python \ -DPYTHON3_LIBRARY=/home/_/miniconda3/envs/newtorch/lib/libpython3.8.so \ -DPYTHON3_INCLUDE_DIR=/home/_/miniconda3/envs/newtorch/include/python3.8 \ -DPYTHON3_NUMPY_INCLUDE_DIRS=/home/_/miniconda3/envs/newtorch/lib/python3.8/site-packages/numpy/core/include \ -DPYTHON3_PACKAGES_PATH=/home/_/miniconda3/envs/newtorch/lib/python3.8/site-packages ..

1. OpenCV要配置CUDA，必须加上contrib，否则也会提示要加上其中一个模块的
2. 一些模块我用不到，比如xfeatures2d, stereo, face但是会下载一些配置文件，特别慢，干脆不编译这个模块了
3. 一些模块包如果要下载的话，可以在build的文件夹里找到CMakeDownloadLog.txt，这个文件记录了下载的日志，包括缓存文件夹位置（opencv/.cache, ls -a命令查看）和下载安装包的链接。
4. 我不是用的系统python环境，需要设置自己的python环境，比如PYTHON_LIBRARY等，[1]里给出了一个比较简单的方法，通过python命令输出环境信息，但是直接用在cmake命令中容易出错，而且PYTHON_LIBRARY也设置错误了，要具体指向python3.8.so，但是根据1里的设置，PYTHON_LIBRARY只指向了那个文件夹，所以会检测不出来，但是可以直接在命令行执行1给的命令得到环境信息，再根据[2]给的例子把自己的环境直接加在cmake里。
5. OpenCV编译、安装完，如果python中import cv2能成功，就代表没什么问题了，但是VScode可能没有代码提示（Pycharm我不知道）,网上也有一些解决方法但是不好使。可以直接找到env_name/lib/python3.8/site-packages/cv2/python-3.8/cv2.cpython-38-x86_64-linux-gnu.so复制到上一个目录，即和__init__.py放在一起，然后在__init__.py中加入from .cv2 import *。当然不复制直接光增加import文件目录到.so的文件夹应该也是可以的。