win10下对编译完成后opencv_cuda进行移植

本教程详细介绍了如何在Windows 10环境下,使用Visual Studio 2015 x64编译OpenCV 4.5.0源码以支持CUDA 10.0,并将编译后的opencv_cuda移植到Python 3.7.4环境中。配置过程包括复制安装文件,将cv2模块放入conda虚拟环境,以及修改配置文件。最后,通过测试代码验证了CUDA加速功能在opencv-python中的成功应用。
部署运行你感兴趣的模型镜像

系列文章目录

win10下Opencv源码编译支持CUDA加速的Python环境,超级详细教程!
win10下对编译完成后opencv_cuda进行移植


接着上一章win10下Opencv源码编译支持CUDA加速的Python环境,超级详细教程!,本章对编译完成后opencv_cuda进行移植
设置适配环境为:
Win10
Visual Studio 2015 x64
Opencv 4.5.0
Cuda10.0
Cudnn7.6.5
Python3.7.4
Numpy1.16.5

一、opencv-python环境配置

1.将前文生成的install 复制粘贴到本地;
2.将前文生成的cv2文件夹复制到所需的conda虚拟环境中;
在这里插入图片描述

3.更改cv2下的配置文件
config.py
在这里插入图片描述
config-3.7.py
在这里插入图片描述

二、 opencv-python cuda加速测试

在所对应的环境下测试CV2是否能够成导入;
在这里插入图片描述
在原来的推理代码中加入使用GPU加速代码;

net.setPreferableBackend(cv2.dnn.DNN_BACKEND_CUDA)
net.setPreferableTarget(cv2.dnn.DNN_TARGET_CUDA)

在这里插入图片描述
如果阅读本文对你有用,欢迎关注点赞评论收藏呀!!!
2021年3月6日10:59:25

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

cmake_minimum_required(VERSION 3.18) project(DFIAlgo LANGUAGES CXX CUDA) # 设置C++标准 set(CMAKE_CXX_STANDARD 17) set(CMAKE_CXX_STANDARD_REQUIRED ON) set(CMAKE_CUDA_STANDARD 17) set(CMAKE_CUDA_STANDARD_REQUIRED ON) # 设置 CUDA 架构 set(CMAKE_CUDA_ARCHITECTURES 61) # 简化 CUDA 编译选项 #set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -arch=sm_61 --expt-relaxed-constexpr") # 查找 CUDA find_package(CUDAToolkit REQUIRED) # 分别处理源文件 file(GLOB HEADER_FILES "*.h" "*.hh" "*.hpp" "../include/*.h") file(GLOB CPP_SRC_FILES "*.cpp" "*.cc") file(GLOB CUDA_SRC_FILES "*.cu") message(STATUS "********C++ head files: ${HEADER_FILES}") message(STATUS "********C++ source files: ${CPP_SRC_FILES}") message(STATUS "********CUDA source files: ${CUDA_SRC_FILES}") # 添加 add_library(DFIAlgo SHARED ${HEADER_FILES} ${CPP_SRC_FILES} ${CUDA_SRC_FILES} ) # 关键:显式设置 CUDA 文件属性 set_source_files_properties(${CUDA_SRC_FILES} PROPERTIES LANGUAGE CUDA) set_target_properties(DFIAlgo PROPERTIES VERSION 1.0.0 SOVERSION 0 CUDA_RUNTIME_LIBRARY Shared ) # 设置文件(从父目录获取或重新定义) if(CMAKE_BUILD_TYPE STREQUAL "Debug") file(GLOB OpenCV_DEBUG_LIBS "C:/Tools/lean/opencv4_5cuda/lib/opencv_*455d.lib") set(OpenCV_LIBS ${OpenCV_DEBUG_LIBS}) else() file(GLOB OpenCV_RELEASE_LIBS "C:/Tools/lean/opencv4_5cuda/lib/opencv_*455.lib") set(OpenCV_LIBS ${OpenCV_RELEASE_LIBS}) endif() list(APPEND OpenCV_LIBS "C:/Tools/lean/opencv4_5cuda/lib/opencv_cudacodec455.lib" "C:/Tools/lean/opencv4_5cuda/lib/opencv_cudaarithm455.lib" "C:/Tools/lean/opencv4_5cuda/lib/opencv_cudaimgproc455.lib" "C:/Tools/lean/opencv4_5cuda/lib/opencv_cudawarping455.lib" ) set(LibTorch_LIBS "C:/Tools/lean/libtorch/lib/c10.lib" "C:/Tools/lean/libtorch/lib/torch_cpu.lib" "C:/Tools/lean/libtorch/lib/torch.lib" ) # 包含目录 target_include_directories(DFIAlgo PRIVATE ${CMAKE_CURRENT_SOURCE_DIR} ${CMAKE_CURRENT_SOURCE_DIR}/../include "C:/Tools/lean/opencv4_5cuda/include" "C:/Tools/lean/libtorch/include" "C:/Tools/lean/libtorch/include/torch/csrc/api/include" "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.5/include" ) target_link_libraries(DFIAlgo ${OpenCV_LIBS} ${LibTorch_LIBS} CUDA::cudart ) 有什么问题?
最新发布
11-26
评论 2
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

AI小笔记

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值