准备CUDA环境,参考教程:
VS2017
cmake编译器
根据自己的Python版本在这里选择合适的OpenCV版本。我的python版本为3.6,选择opencv版本为4.4.0
在这里下载Opencv源码:
在这里下载Opencv_contrib源码:
OpenCV的核心库,就是通常所说的OpenCV。
在核心库之外,还有一个分离开的包含OpenCV最新功能实现的opencv_contrib,也可以称之为OpenCV贡献库。
在Opencv源码目录的CMakeLists.txt
的同级目录下新建目录build
文件夹。
全都通过的话,证明已经安装好了,可以通过c++代码测试是否能够获得gpu。
接下来做的是将源码链接到python环境
进入opencv文件中刚刚建立的build文件夹中,其中有一个python_loader文件夹
执行命令:python setup.py install
卸载环境已有的opencv-python,然后才能使用带cuda的opencv
import cv2