转载：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
来源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。