关于 win11 下 opencv-python GPU 安装构建并读取rtsp视频流（CUDA 12.0.1）

北阿浔

已于 2024-02-25 23:27:38 修改

阅读量3.1k

点赞数 9

CC 4.0 BY-SA版权

文章标签： opencv python 计算机视觉

于 2023-05-30 09:46:20 首次发布

本文链接：https://blog.csdn.net/qq_52318356/article/details/130940884

本文介绍了在Windows11环境下，使用Python3.9.10、CUDA12.0.1和RTX3060GPU，如何解决OpenCV4.7.0DLL加载失败的问题。通过添加DLL目录到环境变量，成功运行RTSP流读取示例，实现GPU加速视频处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考

1. 我受启发的网站
 2. cuda 官方解决方案

我的配置环境

windows 11
python 3.9.10
cuda 12.0.1
RTX 3060

opencv 4.7.0

# 注意，安装 GPU 版的 openvc 之前需要把原本的 opencv 删除

配置过程

下载预构建代码： opencv_contrib_cuda_4.7.0_win_amd64，并解压，得到两个文件夹。

将下面路径添加到 环境变量 path 中：
将 .pyd 文件添加到自己 python 环境下的 site-packages 中即可：可系统、可虚拟

然后你运行以下代码时会报错：

import cv2
# import cv2 as cv
print(f'OpenCV: {cv2.__version__} for python installed and working')


# ImportError: DLL load failed while importing cv2:找不到指定的模块。

在这里插入图片描述

解决方法：你可以添加以下三行：（需要根据自己的安装环境添加）

import os
# 使用os模块中的add_dll_directory函数
# 用于将指定路径添加到DLL搜索路径中，以便在程序运行时加载DLL文件
os.add_dll_directory('C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v12.0\\bin')
os.add_dll_directory('E:\\opencv_contrib_cuda_4.7.0_win_amd64\\install\\x64\\vc17\\bin')

import cv2
# import cv2 as cv
print(f'OpenCV: {cv2.__version__} for python installed and working')

YES，这就成功了

rtsp GPU读取实例（注：这是我抄的，忘记抄谁的了）

import os

os.add_dll_directory('C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v12.0\\bin')
os.add_dll_directory('E:\\opencv_contrib_cuda_4.7.0_win_amd64\\install\\x64\\vc17\\bin')


import cv2

# RTSP URL，修改成自己的URL
rtsp_url = "rtsp://127.0.0.1:8554/video"

# 使用GPU加速
cv2.cuda.setDevice(0)

# 创建VideoCapture对象
cap = cv2.VideoCapture(rtsp_url)

# 检查摄像头是否打开
if not cap.isOpened():
    print("无法打开RTSP流")
else:
    while True:
        # 从RTSP流中读取一帧
        ret, frame = cap.read()

        if ret:
            cv2.imshow('RTSP Stream', frame)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

cap.release()
cv2.destroyAllWindows()