VS2022+libtorch配置

配置选项:基于Windows系统,用VScode+minGW64+cmake+Libtorch(不可行,因为pytorch官网下载的Libtorch是基于MSVC编译的,他与minGW64不兼容),非常感谢这篇博文,减少了绕行时间:基于Windows系统,用VScode+minGW64+cmake+Libtorch开发
解决方法:基于Windows系统,使用VS 2022+Libtorch(首选,后边详述该方法)

步骤一:libtorch下载(DEBUG版本)
前往libtorch官网https://pytorch.org/get-started/locally/,选择对应版本的libtorch进行下载
解压完成后,放入D盘或者其他位置,记录其路径即可:

步骤二:VS2022+libtorch环境配置 [项目] -> [属性]
新建console(控制台)项目,在项目中进行如下配置:
1. 配置include目录: [C/C++]-> [常规]->[附加包含目录]
[你的libtorch目录]\include
[你的libtorch目录]\include\torch\csrc\api\include

2. 配置C++语言标准:[C/C++]-> [语言]
C++语言标准:ISO C++17标准(/std:c++17)

3. 配置linker(链接器):[链接器] -> [常规] -> [附加库目录],添加如下一行:
[你的libtorch目录]\lib\*.lib

4. 配置运行环境:[调试] -> [环境]
PATH=[你的libtorch目录]\lib; %PATH%

测试代码1如下(更多案例可查询网站PyTorch C++ API — PyTorch main documentation):

#include<torch/torch.h>
#include<torch/script.h>
#include<iostream>
using namespace std;
int main()
{
    torch::Tensor output;
    std::cout << "cuda is_available: " << torch::cuda::is_available() << std::endl;
    torch::DeviceType device = at::kCPU;
    if (torch::cuda::is_available())
        device = at::kCUDA;
        output = torch::eye(3).to(device);  //randn({ 3,3 })
        std::cout << output << std::endl;
    return 0;
}

运行结果:

cuda is_available: 0
 1  0  0
 0  1  0
 0  0  1
[ CPUFloatType{3,3} ]

测试代码2:

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

int main() {
    // 使用arange构造一个一维向量,再用reshape变换到5x5的矩阵
    torch::Tensor foo = torch::arange(25).reshape({ 5, 5 });

    // 计算矩阵的迹
    torch::Tensor bar = torch::einsum("ii", foo);

    // 输出矩阵和对应的迹
    std::cout << "==> matrix is:\n " << foo << std::endl;
    std::cout << "==> trace of it is:\n " << bar << std::endl;
}

运行结果:

==> matrix is:
   0   1   2   3   4
  5   6   7   8   9
 10  11  12  13  14
 15  16  17  18  19
 20  21  22  23  24
[ CPULongType{5,5} ]
==> trace of it is:
 60
[ CPULongType{} ]

参考资料

PyTorch C++ API — PyTorch main documentation

Libtorch系列教程1:一个丝滑的C++ Tensor库-腾讯云开发者社区-腾讯云 (tencent.com)

  • 6
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
要在VSCode中添加Boost库,可以按照以下步骤进行操作: 1. 安装Boost库:按照引用中提供的教程,下载并安装Boost库。 2. 打开VSCode,并打开你的项目文件夹。 3. 找到项目文件夹中的`.vscode`文件夹,如果没有,请手动创建一个。 4. 在`.vscode`文件夹中创建一个名为`settings.json`的文件。 5. 在`settings.json`文件中添加以下内容:`"-I F:\\Library\\boost_1_78_0\\"`。这将告诉编译器在指定路径中查找Boost库的头文件。 6. 保存`settings.json`文件。 7. 打开项目文件夹中的`tasks.json`文件(如果没有,请手动创建一个)。 8. 在`tasks.json`文件中,在`args`数组中添加以下参数: ``` "${file}", "-o", "${fileDirname}/${fileBasenameNoExtension}.exe", "-g", "-I F:\\Library\\boost_1_78_0\\", "-Wall", "-static-libgcc", "-fexec-charset=UTF-8", "-std=c 17", "-lpthread", "-lboost_thread", "-lboost_system" ``` 这将告诉编译器在指定路径中查找Boost库的库文件并链接它们。 9. 保存`tasks.json`文件。 现在,你已经成功在VSCode中添加了Boost库。编译和运行你的代码时,编译器将使用指定路径中的Boost库文件。 引用: : 需要添加 "-I F:\\Library\\boost_1_78_0\\", -I后的地址是安装Boost的位置。`setttings.json`。 : 安装教程可参考:VSCode安装教程(超详细版)_8号葫芦娃的博客-CSDN博客_vscode安装 二、Boost下载。 : 在`tasks.json`文件中`"args": [ "${file}", "-o", "${fileDirname}/${fileBasenameNoExtension}.exe", "-g", "-I F:\\Library\\boost_1_78_0\\", "-Wall", "-static-libgcc", "-fexec-charset=UTF-8", "-std=c 17", "-lpthread", "-lboost_thread", "-lboost_system" ]。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值