1、在https://pytorch.org中下载对应版本的libtorch。
2、解压后于环境变量的path中添加xxx\libtorch\lib。
3、打开vs2019建立新项目,选择release和x64模式,属性管理器中在release|x64中添加新项目属性表起名libtorch(随便起一个)。
4、包含目录添加:
xxx\libtorch\include
xxx\libtorch\include\torch\csrc\api\include
5、库目录添加:
xxx\libtorch\lib
6、链接器输入添加:
asmjit.lib
c10.lib
caffe2_detectron_ops.lib
caffe2_module_test_dynamic.lib
clog.lib
cpuinfo.lib
dnnl.lib
fbgemm.lib
libprotobuf.lib
libprotobuf-lite.lib
libprotoc.lib
mkldnn.lib
torch.lib
torch_cpu.lib
(即xxx\libtorch\lib中的所有lib文件,不同版本的libtorch不一样,可能不全需要,保险起见可以全加上)
**配置opencv的方法类似
7、相关头文件为:
#include <torch/torch.h>
#include <torch/script.h>
8、可使用以下代码测试libtorch是否正常导入:
torch::Tensor tensor = torch::rand({2, 3});
std::cout << tensor << std::endl;
**若运行出现问题,将xxx\libtorch\lib中的所有dll文件放到vs工程的exe同级目录中再尝试调试。