安装环境

CUDA Toolkit

注意与显卡驱动版本相匹配

cuDNN

注意CUDA版本
将压缩包里面的三个文件夹放入CUDA Toolkit安装目录，默认路径

1
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7

下载预构建的opencv源代码

预构建的opencv源代码

opencv_4_5_0_cuda_11_1_py38.7z文件解压后，得到两个文件夹，install和lib

install文件夹，依次进入路径install\x64\vc16\bin，然后添加将其到系统环境变量中

lib\python3\cv2.cp38-win_amd64.pyd文件复制放入自己python环境的site-packages中

查看已安装的opencv是否支持cuda

1
2
import cv2
print(cv2.cuda.getCudaEnabledDeviceCount())

如果返回值大于0，则说明OpenCV已经安装了CUDA支持