在Jetson Orin Nano上编译支持cuda的opencv-python

环境

• Jetson Orin Nano 8GB
• Cuda 12.2
• Cudnn 8.9
• Jetpack 6.0 DP
• python 3.11
• miniconda

编译过程

拉取源码

git clone --recursive https://github.com/opencv/opencv-python.git
使用这条命令会自动拉取源码和全部依赖

编译

CMAKE_ARGS="-DBUILD_ZLIB=ON -DBUILD_TIFF=ON -DBUILD_OPENJPEG=ON -DBUILD_JASPER=ON -DBUILD_JPEG=ON -DBUILD_PNG=ON -DBUILD_OPENEXR=ON -DBUILD_WEBP=ON -DBUILD_TBB=ON -DBUILD_ITT=ON -DWITH_1394=ON -DWITH_AVIF=ON -DWITH_CAROTENE=ON -DWITH_VTK=ON -DWITH_CUDA=ON -DWITH_CUFFT=ON -DWITH_CUBLAS=ON -DWITH_CUDNN=ON -DWITH_EIGEN=ON -DWITH_FFMPEG=ON -DWITH_GSTREAMER=ON -DWITH_WAYLAND=ON -DWITH_HALIDE=ON -DWITH_VULKAN=ON -DWITH_JASPER=ON -DWITH_OPENJPEG=ON -DWITH_JPEG=ON -DWITH_WEBP=ON -DWITH_OPENEXR=ON -DWITH_OPENGL=ON -DWITH_OPENVX=ON -DWITH_OPENNI=ON -DWITH_OPENNI2=ON -DWITH_PNG=ON -DWITH_SPNG=ON -DWITH_GDCM=ON -DWITH_PVAPI=ON -DWITH_ARAVIS=ON -DWITH_QT=ON -DWITH_TBB=ON -DWITH_OPENMP=ON -DWITH_PTHREADS_PF=ON -DWITH_TIFF=ON -DWITH_V4L=ON -DWITH_XIMEA=ON -DWITH_UEYE=ON -DWITH_XINE=ON -DENABLE_FLAKE8=ON -DENABLE_PYLINT=ON -DENABLE_CUDA_FIRST_CLASS_LANGUAGE=ON -DOPENCV_SEMIHOSTING=ON -DENABLE_CCACHE=ON -DENABLE_FAST_MATH=ON -DWITH_FLATBUFFERS=ON -DWITH_CANN=ON -DDWITH_GTK=ON -DOPENCV_EXTRA_MODULES_PATH='$PWD/opencv_contrib/modules' -DOPENCV_ENABLE_MEMORY_SANITIZER=ON -DCUDA_FAST_MATH=ON" python setup.py build
cmake中的参数根据自己的需要进行修改，可以参考这里。需要安装部分依赖库，请自行搜索安装。

打包

opencv-python.whl:python setup.py bdist_wheel
opencv-contrib-python.whl ENABLE_CONTRIB=1 python setup.py bdist_wheel
opencv-python-headless.whl ENABLE_HEADLESS=1 python setup.py bdist_wheel
打包后的文件在dist目录下

安装

pip install 即可