
libtorch
文章平均质量分 81
CVer儿
开源让世界更美好,闭源让人更富裕
展开
-
libtorch的混合精度使用
手动缩放损失是混合精度训练中防止梯度下溢的关键步骤,1024 是一个经验性安全值,需根据实际训练效果调整。原创 2025-04-11 00:59:40 · 589 阅读 · 0 评论 -
torchvision编译血泪史二
相关方案都说pip安装对应库,测试无效,因此从头开始编译zlib和libPng。版本采用libtorch:2.3.1 torchvision 0.18.1。采用cmae gui编译边都是 win32,因此采用命令行编译。原创 2024-11-14 14:02:59 · 714 阅读 · 0 评论 -
ONNX那些事
ONNX (Open Neural Network Exchange)是 Facebook 和微软在2017年共同发布的,用于标准描述计算图的一种格式。目前,在数家机构的共同维护下,ONNX 已经对接了多种深度学习框架和多种推理引擎。因此,ONNX 被当成了深度学习框架到推理引擎的桥梁,就像编译器的中间语言一样。由于各框架兼容性不一,我们通常只用 ONNX 表示更容易部署的静态图。原创 2024-09-23 16:40:53 · 1032 阅读 · 0 评论 -
libtorch 模型dectypt
这回v_ch也有2千万的数据了,大小也是一样的,可以torch还是报错!这个小功能花费了一天半的时间,其实我一开始的半天就搞通可以流读到load里面,然后都是在解决上面的问题,std::filebuf的那个问题还不知道哪里有问题!一般我们都是用上面这个,直接给模型路径就可以了,下面这个是流,没用过。在python强大的第三方插件库中,就有这么一个插件,很好的解决了我们的问题,这个插件的名为:cryptography。这样,正常的和不正常的都保存,然后比较两个文本,发现不正常的最后比正常的多了一个数字。转载 2023-07-14 10:16:56 · 275 阅读 · 0 评论 -
libtorch 调用模型方法
libtorch 是 pytorch 的 C++ 版本,支持 CPU 端和 GPU 端的模型部署。相较于用 tensorrt 部署 pytorch 模型,用 libtorch的优势在于:pytorch 和 libtorch 同属一个生态,API 语句比较接近,并且不会出现某网络层不支持的问题。链接:https://juejin.cn/post/7153132249028755463。这里就会导出 trace 模型 resnet50.pt。首先加载 resnet50 模型,并导出 trace 模型。原创 2023-07-13 17:52:24 · 781 阅读 · 0 评论 -
Libtorch win10实现人像背景抠图
高版本VS中有个bug,需要手动添加点东西我们的代码才能正常的调用GPU。找到链接器——命令行——其它选项,然后输入: /INCLUDE:?转载 2023-07-13 17:42:28 · 185 阅读 · 0 评论 -
LibTorch | 求解一维稳态对流扩散方程
本文将使用基于LibTorch接口)的神经网络求解器,对一维稳态对流扩散方程进行求解,文中仅对神经网络求解器对特定问题的求解能力进行了介绍,单纯一个无监督学习问题,未涉及迁移到其他问题的适用性问题等。水平有限,如有问题还希望读者斧正。研究问题参考自教科书[1][1]示例 8.3。目录。转载 2023-07-13 17:20:27 · 937 阅读 · 0 评论 -
cuda数据初始化libtorch tensor
上述代码中,首先在主机内存中初始化了一个大小为$n \times m$的数据矩阵。方法,而不是普通的加法运算符。同样的,可以使用其他GPU计算支持的函数和方法来完成操作。通过将CPU内存中的数据指针和GPU显存中的数据指针传递给。函数将Tensor对象转换为主机内存中的数组,并打印结果。需要注意的是,在GPU端对数据进行计算和操作时,需要使用GPU上支持的。函数将GPU内存中的数据构造为Tensor对象,然后使用。函数,将数组从主机内存复制到GPU内存中。例如,在上面的例子中,使用的是。原创 2023-05-06 18:10:34 · 1010 阅读 · 0 评论 -
libtorch无法启动gpu
一直设备为cpu,参考。原创 2023-03-24 17:36:17 · 432 阅读 · 1 评论 -
libtorch Tensor张量的常用操作
定义一定维度的张量并初始化全部值为0//定义5行7列的0值张量定义一定维度的张量并初始化全部值为1//定义3行4列的1值张量定义一定维度的单位张量//定义5*5单位张量定义一定维度的张量并设置初始值//定义3行4列张量,并初始化全部值为0运行结果如下,得到3行4列的张量://定义3行4列的张量b,并填充全部值为10//定义与b相同形状的张量a,并填充初始值2//定义与b相同形状的张量a,并填充初始值2.5//定义与b相同形状的张量a,并填充初始值2.5。转载 2023-01-03 13:32:25 · 1036 阅读 · 1 评论