确定安装好了cmake,nvidia-driver,Cuda,cudnn
在ubuntu中编译好对应的ffmpeg
准备将python版本的opencv,编译到conda的环境中,提前创建好自己的conda虚拟环境,我这里是cd_test, 也可以编译到其他的python环境中,具体路径得配置正确。
下载opencv4.5.4
官方源码地址
下载opencv-contribute4.5.4
可以在此下载国内镜像
解压到同一目录下,结构为:
folder
|–opencv-4.5.4
|–opencv_contrib-4.5.4
mkdir build cd build #查看cmake选项 cmake -LH .. # 配置cmake选项,带GPU cmake -D CMAKE_BUILD_TYPE=RELEASE \ -D CMAKE_INSTALL_PREFIX=../linux_install \ -D INSTALL_C_EXAMPLES=OFF \ -D INSTALL_PYTHON_EXAMPLES=OFF \ -D OPENCV_ENABLE_NONFREE=OFF \ -D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib-4.5.4/modules \ -D WITH_FFMPEG=on \ -D WITH_CUDA=ON \ -D WITH_CUBLAS=ON \ -D CUDA_FAST_MATH=ON \ -D WITH_CUFFT=ON \ -D WITH_NVCUVID=ON \ -D WITH_V4L=ON \ -D WITH_LIBV4L=ON \ -D WITH_OPENGL=ON \ -D BUILD_EXAMPLES=OFF \ -D OPENCV_GENERATE_PKGCONFIG=YES \ -D BUILD_opencv_world=NO \ -D BUILD_opencv_python2=OFF \ -D BUILD_opencv_python3=ON \ -D PYTHON_DEFAULT_EXECUTABLE=/home/hw/anaconda3/envs/cd_test/bin/python3 \ -D PYTHON3_EXECUTABLE=/home/hw/anaconda3/envs/cd_test/bin/python3 \ -D PYTHON3_LIBRARY=/home/hw/anaconda3/envs/cd_test/lib/libpython3.8.so \ -D PYTHON_INCLUDE_DIR=/home/hw/anaconda3/envs/cd_test/include/python3.8 \ -D PYTHON3_NUMPY_INCLUDE_DIRS=/home/hw/anaconda3/envs/cd_test/lib/python3.8/site-packages/numpy/core/include \ -D PYTHON3_PACKAGES_PATH=/home/hw/anaconda3/envs/cd_test/lib/python3.8/site-packages \ .. # ffmpeg版本 cmake -D CMAKE_BUILD_TYPE=RELEASE \ -D CMAKE_INSTALL_PREFIX=../linux_install \ -D INSTALL_C_EXAMPLES=OFF \ -D INSTALL_PYTHON_EXAMPLES=OFF \ -D OPENCV_ENABLE_NONFREE=OFF \ -D WITH_FFMPEG=ON \ -D BUILD_EXAMPLES=OFF \ -D OPENCV_GENERATE_PKGCONFIG=YES \ -D BUILD_opencv_world=OFF \ -D BUILD_opencv_python2=OFF \ -D BUILD_opencv_python3=ON \ -D PYTHON_DEFAULT_EXECUTABLE=/home/hw/anaconda3/envs/cd_test/bin/python3 \ -D PYTHON3_EXECUTABLE=/home/hw/anaconda3/envs/cd_test/bin/python3 \ -D PYTHON3_LIBRARY=/home/hw/anaconda3/envs/cd_test/lib/libpython3.8.so \ -D PYTHON_INCLUDE_DIR=/home/hw/anaconda3/envs/cd_test/include/python3.8 \ -D PYTHON3_NUMPY_INCLUDE_DIRS=/home/hw/anaconda3/envs/cd_test/lib/python3.8/site-packages/numpy/core/include \ -D PYTHON3_PACKAGES_PATH=/home/hw/anaconda3/envs/cd_test/lib/python3.8/site-packages \ ..
说明:如果该命令中不加-D CMAKE_INSTALL_PREFIX,则默认各部分分别安装在/usr/local/目录的include/, bin/, lib/3个文件夹下。
-D OPENCV_GENERATE_PKGCONFIG=YES: OpenCV4以上版本默认不使用pkg-config,该编译选项开启生成opencv4.pc文件,支持pkg-config功能。
-D WITH_CUDA=ON实现和cuda的联合编译。
后面关于python的路径设置以自己机子anaconda的设定为准,需要注意的是python版本和你使用的虚拟环境一致。
【问题1】fatal error: ImfChromaticities.h: No such file or directory
【解决办法】 sudo apt-get install libopenexr-dev
【问题2】ffmpeg选项是no
-- OpenCV Python: during development append to PYTHONPATH: /dataset/users/chengd/envs/opencv/opencv-4.5.4/build/python_loader -- WARNING: Can't build ffmpeg test code
【解决办法】首先记得删除build文件夹中的缓存文件,重新cmake,不然会还是no
其次cmake过程会打印no的原因,比如我这里是so没找着
最后发现编译生成的/usr/local/ffmpeg3.4.9/lib/pkgconfig/libavcodec.pc中的路径是错的,
给改到正确位置
同时记得删除build文件夹中的缓存文件,不然会还是no。
解决完问题,就可以顺利找到了
【问题3】编译时ffmpeg的某些函数找不着
【解决办法】最后发现时ffmpeg的版本问题,ffmpeg最新版很多方法更新移除了,目前opencv4.x版本编译与ffmpeg3.x版本吻合。所以这里编译了ffmpg3.4.9版本
make -j6 sudo make install
conda activate cd_test #此时在conda和pip的包管理器中都没有opencv-python,但是可以运行 python >>> import cv2 >>> print(cv2.__version__) 4.5.4
可能提示你先要安装numpy,装一下就好了pip install numpy