随笔-376  评论-37  文章-0  trackbacks-0
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 小王 阅读(372) 评论(0)  编辑 收藏 引用 所属分类: LibTorch

只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理