libtorch在windows上的调用

转载：https://www.jianshu.com/p/542231f43207

libTorch下载

pytroch为windows端提供了2个版本的预编译好的libTorch动态链接库

Debug

Release

image.png

---

测试环境

• win10 64bit
• vs2017
• libTorch

---

配置过程

以Debug版本的libTorch为例

• 添加include路径
• 添加链接库lib路径
• 添加lib名称
• 添加环境变量

image.png

image.png

image.png

c10.lib
caffe2.lib
caffe2_detectron_ops.lib
caffe2_module_test_dynamic.lib
clog.lib
cpuinfo.lib
foxi_dummy.lib
foxi_loader.lib
libprotobuf-lited.lib
libprotobufd.lib
libprotocd.lib
onnx.lib
onnxifi_dummy.lib
onnxifi_loader.lib
onnx_proto.lib
torch.lib

---

测试代码

• demo1

#include<iostream>
#include<torch/script.h>
 
int main() {
 
    torch::Tensor t1 = torch::tensor({ 10,1,2 });
    std::cout << t1[0] << std::endl;
 
    system("pause");
 
}

image.png

• demo2
  2个Tensor计算矩阵乘法

#include<iostream>
#include<torch/script.h>
 
int main() {
 
    
    auto t1 = torch::tensor({ 1,2,3,4,5,6,7,8,9 }).reshape({ 3,3 });
    auto t2 = torch::tensor({ 1,0,2,6,1,1,5,3,2 }).reshape({ 3,3 });
 
    auto t3 = t1.mul(t2);
 
    std::cout << t3 << std::endl;
    
 
    system("pause");
 
}

image.png

作者：侠之大者_7d3f
链接：https://www.jianshu.com/p/542231f43207
来源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。