VS2017配置libtorch:https://blog.csdn.net/weixin_41449637/article/details/109812646
-
1. Libtorch下载官网:
https://pytorch.org/get-started/locally/
https://blog.csdn.net/weixin_43742643/article/details/115218126
注意:Release和Debug版本,我下载的是debug版本
-
2、在VS2019中的配置
需要配置.头文件h 库文件,lib, 和dll文件。
配置头文件,按以下步骤操作
在步骤4,依次添加:
*** \libtorch\include
*** \libtorch\include\torch\csrc\api\include
配置依赖库的位置。
添加依赖库的文件名。
asmjit.lib c10.lib c10_cuda.lib caffe2_detectron_ops_gpu.lib caffe2_module_test_dynamic.lib caffe2_nvrtc.lib Caffe2_perfkernels_avx.lib Caffe2_perfkernels_avx2.lib Caffe2_perfkernels_avx512.lib clog.lib cpuinfo.lib dnnl.lib fbgemm.lib fbjni.lib kineto.lib libprotobufd.lib libprotobuf-lited.lib libprotocd.lib mkldnn.lib pthreadpool.lib pytorch_jni.lib torch.lib torch_cpu.lib torch_cuda.lib XNNPACK.lib
设置dll文件的环境变量。
(dll文件拷贝到生成的exe文件所在目录,导致这些dll文件会复制很多份,占据很大空间,这里通过设置依赖于项目的环境变量,少了大量拷贝,还不影响其他变量)
PATH= F:\Pycharm\PyCharm_Study\Others\c++_learning\C++_Master\Onnx\libtorch\libtorch-win-shared-with-deps-debug-1.10.0+cu102\libtorch\lib;%PATH%
-
3. 测试
#include <iostream> #include <torch/torch.h> int main() { torch::Tensor tensor = torch::rand({ 5,3 }); std::cout << tensor << std::endl; printf("Done!\n"); system("pause"); return 0; }
测试成功!
参考:https://yanfeng2012.blog.csdn.net/article/details/106481312