pytorch相关
文章平均质量分 88
pytorch学习以及bug记录
我宿孤栈
略略略~
展开
-
VS编译部署libtorch-yolov5推理运行自己训练的权重文件/模型(CPU和GPU版本)
前期环境配置(vs+libtorch+opencv)可以参考博主另一篇博文,这里主要基于环境配置好之后如何运行yolov5的推理程序,并生成对应的.exe文件。libtroch版本尽量和pytorch的版本一致,。博主查阅了很多博文,有的博文是通过cmake编译运行,博主vs新手,因为VS配置的libtorch和opencv是跟着项目的(在vs界面),博主暂时还没能把终端cmake编译和vs联系起来(配置好的环境容易崩掉),后边有时间再出一版结合cmake编译运行的方案。原创 2024-04-17 16:59:35 · 1560 阅读 · 1 评论 -
win下pytorch安装—cuda11.7 + cudnn8.4 + pytorch1.13安装
这里选择默认目录即可,是个安装过程的临时目录,结束后此文件夹会被自动删除。需要注意的是这里并非指定安装目录,若需要指定安装目录,此处需与指定安装目录不同。即可查看显卡适配的cuda版本,要原创 2023-03-22 15:30:16 · 3733 阅读 · 2 评论