VS2019 配置 LibTorch 和 OpenCV

辛琪

2022-01-28 2022-01-28 约 534 字预计阅读 2 分钟次阅读

下载所需要的库

LibTorch OpenCV

配置环境变量

LibTorch 配置到 lib 目录就好，OpenCV根据自己选择使用的 vc14 或 vc15 配置到其下面的 bin 目录

用VS新建一个空项目

配置 OpenCV

右键项目选择属性

\2. 更改配置为所有配置并选择 VC++ 目录中的包含目录

\3. 添加头文件到包含目录 (刚才下载的OpenCV目录) opencv/build/include

\4. 添加库目录并保存

\5. 添加 OpenCV 库文件名字到链接器-输入-附加依赖项，其中添加的名字和刚才配置的库目录中的lib文件名相同

\6. 在源文件添加一个文件测试OpenCV

\7. 添加如下代码进行测试，更改运行环境为 x64 点击运行

1
2
3
4
5
6
7


#include <opencv2/opencv.hpp>
int main() {
	cv::Mat img = cv::imread("图片路径");
	cv::imshow("", img);
	cv::waitKey(0);
	return 0;
}

配置 LibTorch

在属性-VC++目录-包含目录中添加 LibTorch 头文件

\2. 在属性-VC++目录-库目录中添加 LibTorch 的库目录

\3. 在属性-链接器-输入-附加依赖项中添加LibTorch的依赖

1
2
3
4


c10.lib
torch.lib
torch_cpu.lib
torch_cuda.lib

\4. 添加测试代码测试

1
2
3
4
5
6
7
8


#include <torch/torch.h>
#include <iostream>
int main() {
	torch::Tensor data = torch::rand({ 3, 3 });
	std::cout << data << std::endl;
	std::cout << torch::cuda::is_available() << std::endl;
	return 0;
}

我机器上面是有 cuda 的但是这里显示没有 cuda，下面进行 cuda 配置可能有些版本不同

\5. 配置支持 cuda

在属性-链接器-命令行中添加如下字符串

1
2
3
4
5


# 下面是 cuda10 的
/INCLUDE:?warp_size@cuda@at@@YAHXZ

# cuda11 可以尝试下面的
/INCLUDE:?searchsorted_cuda@native@at@@YA?AVTensor@2@AEBV32@0_N1@Z 

测试

可能出现问题的解决方案

没有配置环境变量

属性-调试-环境中添加环境变量（换成自己的地址）

1

PATH=\path\to\libtorch\lib;\path\to\opencv\build\x64\vc15\bin;%PATH%;