假设我现在已经训练好了一个模型,效果挺好,想用C++调。流程就是这么个流程:
- 配置libtorch
- pytorch模型转化
- 编写C++调用程序
这里就先看一下第一步。怎么配置libtorch。
首先去官网下载
PyTorch https://pytorch.org/
解压后是这个样子
使用方法呢和一般的库一样
包含目录两个
库目录一个
链接器->输入->附加依赖项,这里刚开始没写,一堆错误。干脆写全了
就是libtorchlib目录下的所有lib文件名
然后就行了。
测试一下
#include#include int main() { torch::Tensor tensor = torch::rand({ 5,3 }); std::cout << tensor << std::endl; return 0; }



