1、下载libtorch库
https://pytorch.org/
根据提示选项选择下载版本
我选择的是:
stable1.12.0
Windows
LibTorch
C++/Java
CPU
将下载的压缩包解压后放到项目解决方案根目录中
2、导入库文件
(1)include
..\libtorch\include\
..\libtorch\include\torch\csrc\api\include
(2)lib
..\libtorch\lib
以及下面的各种.lib文件
3、测试工程代码
#include <iostream>
#include <torch/torch.h>
int main()
{
torch::Tensor tensor = torch::rand({ 5,3 });
std::cout << tensor << std::endl;
return EXIT_SUCCESS;
}
编译运行!
O了!
posted on 2022-07-05 18:46
小王 阅读(423)
评论(0) 编辑 收藏 引用 所属分类:
LibTorch