C++中使用libtorch

这篇博客介绍了如何在C++环境中使用libTorch调用PyTorch的预训练模型。首先,详细阐述了PyTorch C++环境的搭建过程,包括下载libTorch接口代码并配置CMakeLists.txt。接着,提到了在编译过程中可能出现的错误及其解决方法。此外,还探讨了C++代码中与OpenCV的集成。最后,提供了两个参考资料,分别关于libtorch使用中遇到的问题和常用的libtorch函数。
摘要由CSDN通过智能技术生成
  • pytorch C++接口调用,最简单示例:

pytroch学习(二十一)—C++(libTorch)调用pytroch预训练模型

Pytorch C++ 环境搭建

2个教程类型,步骤:下载pytorch 的C++ 接口代码,解压后即可使用,不用安装。

包括CMakeLists.txt的设置:

cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
project(example-app)

find_package(Torch REQUIRED)

add_executable(example-app example-app.cpp)
target_link_libraries(example-app "${TORCH_LIBRARIES}")
set_property(TARGET example-app PROPERTY CXX_STANDARD 11)

example-app.cpp:

#include <torch/torch.h>
#include <iostream>

int main() {
  torch::Tensor tensor = torch::rand({2, 3});
  std::cout << tensor << std::endl;
}

编译及运行:

cmake .
make
./example-app
  •  一个错误:cmake .的时候找不到TorchConfig.cmaketorch-config.cmake文件, 解决是在CMakeLists.txt中设置,

set( CMAKE_PREFIX_PATH  "/****/libtorch-cxx11-abi-shared-with-deps-1.4.0/libtorch/share/cmake/Torch")#***部分写自己的路。来自关于windows10用c++部署libtorch过程中遇到的一些问题

  • 具体C++中代码相关,包括opencv的调用

2个参考文章:

Libtorch踩坑实录:non-scalar type, '->' has non-pointer type,opencv,Expected Tensor but got Tuple

libt

  • 4
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值