Windows11_yolov5+_C++_Libtorch+VS2019学习记录
由于实验室要求将pytorch的yolov5集成到C++上实现项目落地以及加速所以我第一时间想到了Libtorch
下面就是具体的实施步骤
(本人是一名学生没有太多的知识体系请各位大佬勿喷,谢谢!!)
1.Libtorch下载:
首先,你要先查看自己的yolov5下的torch版本:
打开你的项目文档(我用的是pycharm打开的)新建一个py文件(想怎么命名都可以)然后输入以下代码:
import torch
print(torch.__version__)
右击运行改py文件然后显示出你的torch的版本,例如我就是torch=1.10,cuda=10.2
这部分过程结束之后我们开始去往libtorch官网(其实也就是和pytorch是同一个网址):Libtorch官网
进入后下拉看到该页面选择对应你torch版本的libtorch版本然后选择相对应的cuda版本进行下载然后这个选项里边推荐选release版本的(我选的release)
过去的libtorch版本点这里
(下载很慢各位耐心等待)
下载完成后可以进行OpenCV的环境安装了
2.OpenCV的下载
OpenCV官网链接
我自己是下载的最新版本的OpenCV就直接下windows版本就好了如果有的话就不用下载了
整理文件并记住OpenCV的路径
3.环境变量的配置
打开自己的编辑系统环境变量
然后将Libtorch的lib目录,并将OpenCV的vc15的bin路径添加到用户变量的path中去
4.VS2019的配置
这里我是参照着这篇博客安装的Libtorch的使用总结
参照这个博主的前几个步骤就可以的配置好release
此系列还会继续更新谢谢支持!