在C++中调用Pytorch模型

CPU环境配置

# 下载地址
https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-1.5.0%2Bcpu.zip​
1
2

包含目录

C:\libtorch-win-shared-with-deps-1.5.0+cpu\libtorch\include
C:\libtorch-win-shared-with-deps-1.5.0+cpu\libtorch\include\torch\csrc\api\include
1
2

库目录

C:\libtorch-win-shared-with-deps-1.5.0+cpu\libtorch\lib
1

链接器->输入

asmjit.lib
c10.lib
caffe2_detectron_ops.lib
caffe2_module_test_dynamic.lib
clog.lib
cpuinfo.lib
fbgemm.lib
libprotobuf.lib
libprotobuf-lite.lib
libprotoc.lib
mkldnn.lib
torch.lib
torch_cpu.lib
1
2
3
4
5
6
7
8
9
10
11
12
13

GPU环境配置

# 下载地址
https://download.pytorch.org/libtorch/cu101/libtorch-win-shared-with-deps-1.5.0.zip
1
2

包含目录

C:\libtorch-win-shared-with-deps-1.5.0\libtorch\include
C:\libtorch-win-shared-with-deps-1.5.0\libtorch\include\torch\csrc\api\include
1
2

库目录

C:\libtorch-win-shared-with-deps-1.5.0\libtorch\lib
1

链接器->输入

asmjit.lib
c10.lib
c10_cuda.lib
caffe2_detectron_ops_gpu.lib
caffe2_module_test_dynamic.lib
caffe2_nvrtc.lib
clog.lib
cpuinfo.lib
fbgemm.lib
libprotobuf.lib
libprotobuf-lite.lib
libprotoc.lib
mkldnn.lib
torch.lib
torch_cpu.lib
torch_cuda.lib
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

OpenCV环境配置

包含目录：

D:\opencv-3.4.1-vc14_vc15\opencv\build\include
D:\opencv-3.4.1-vc14_vc15\opencv\build\include\opencv
D:\opencv-3.4.1-vc14_vc15\opencv\build\include\opencv2
1
2
3

库目录：

D:\opencv-3.4.1-vc14_vc15\opencv\build\x64\vc14\lib
1

链接器->输入：

opencv_world341.lib
opencv_world341d.lib
1
2