【C++】VS配置OpenCV/Libtorch环境

zstar

发布于 2023-09-02 18:51:33

3900

发布于 2023-09-02 18:51:33

文章被收录于专栏：往期博文往期博文

前言

本文是视频https://www.bilibili.com/video/BV1dp4y177L4的笔记。

OpenCV和Libtorch安装包：https://pan.baidu.com/s/1i3DqTcHFSC1rRDsIgYGCsQ?pwd=8888

VS版本：2019 Opencv版本：3.4.1 Libtorch版本：2.0.1+cu117

配置OpenCV环境

1.打开VS，创建控制台应用。

2.在视图中，调出属性管理器。

3.属性管理器中，选择Debug|x64的属性。

4.在包含目录中添加OpenCV的路径，我这里是E:\C_Libiary\opencv3.41\build\include和E:\C_Libiary\opencv3.41\build\include\opencv2

5.库目录添加E:\C_Libiary\opencv3.41\build\x64\vc15\lib

6.附加依赖项添加opencv_world341d.lib

7.环境变量添加E:\C_Libiary\opencv3.41\build\x64\vc15\bin

8.将bin目录下的三个dll文件复制到C:\Windows\System32路径中

9，在cpp文件下放置一张图片，调用下面代码进行测试

#include <iostream>
#include <opencv2/opencv.hpp>

using namespace cv;
using namespace std;

int main()
{
    Mat src = imread("1.png"); /*图片 */
    imshow("input", src);
    waitKey(0);
    return 0;
}

成功打开，说明OpenCV环境配置成功。

配置Libtorch环境

1.在包含目录中添加Libtorch的路径，我这里是E:\C_Libiary\libtorch\include\torch\csrc\api\include和E:\C_Libiary\libtorch\include

2.添加库目录，路径为E:\C_Libiary\libtorch\lib

3.在依赖项添加以下内容(Libtorch文件夹下的一些lib文件，不同版本的Libtorch会略有区别)

asmjit.lib
c10.lib
c10_cuda.lib
caffe2_nvrtc.lib
clog.lib
cpuinfo.lib
dnnl.lib
fbgemm.lib
fbjni.lib
kineto.lib
libprotobuf.lib
libprotobuf-lite.lib
libprotoc.lib
nvfuser_codegen.lib
pthreadpool.lib
pytorch_jni.lib
torch.lib
torch_cpu.lib
torch_cuda.lib
XNNPACK.lib

4,添加环境PATH=E:\C_Libiary\libtorch\lib;%PATH%

5.复制所有的dll文件到C:\Windows\System32路径中

6.输入以下代码进行测试

#include<torch/torch.h>
#include<torch/script.h>

using namespace torch;
using namespace std;


int main() {
	Tensor tensor = torch::rand({ 1,2,3 });
	cout << tensor.sizes() << endl;   //方式一，只打印维度信息
	tensor.print();    //方式二，除了打印维度信息，数据类型也打印出来
	return 0;
}