win10 c++通过libtorch调用yolov5模型 配置及使用

本文介绍了在Windows 10环境下,使用Visual Studio 2019和libtorch 1.8.1(CUDA 10.2)调用Yolov5模型的配置和步骤。详细讲解了libtorch、OpenCV 3.4.14的下载、CUDA安装、环境变量设置、VS项目配置,并提供了常见问题的解决方案,如torch_cuda.dll的链接错误和std符号不明确的问题。

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

环境:

windows 10;
vs2019;
libtorch1.8.1-cuda10.2
opencv3.4.14
cmake 3.20.2

前期准备:

libtorch下载(Release版):

https://pytorch.org/get-started/locally/中寻找符合自己的版本,我的选择如下:
在这里插入图片描述
建议使用release version,debug version未实测过。
这里的pytorch版本需和训练yolov5的pytorch版本统一,如果需要其他版本的libtorch,推荐看这篇博客

我使用的版本的地址如下:
gpu:https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-1.8.1%2Bcu102.zip
cpu:https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.8.1%2Bcpu.zip
一般只下载gpu版即可,除非你只需在cpu上跑模型。
下载完成后解压在任意地址即可,这里假定文件在D:\libtorch中。libtorch文件夹内容如下。
目录

opencv下载:

https://opencv.org/releases/中选择自己所需版本即可。我这里选择的是3.4.14的windows版。https://sourceforge.net/projects/opencvlibrary/files/3.4.14/opencv-3.4.14-vc14_vc15.exe/download
双击执行,解压在任意地址即可,这里假定文件在D:\opencv中,文件夹内容如下。
在这里插入图片描述

cuda10.2

下载cuda10.2并安装,这类教程就很多了。安装完成后在cmd中输入nvcc -V,若能成功输入版本号即为成功。
在这里插入图片描述
下载cudnn(最新版本即可),并按教程放在cuda文件夹内。

环境变量设置

安装完成后,右键“此电脑”->“属性”->“高级系统设置”->“环境变量”。对用户变量中的Path进行修改

在这里插入图片描述
加入"D:\libtorch\lib",“D:\opencv\build\x64\vc15\bin”(具体前缀地址以你实际放置的地址为准)

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值