Libtorch的使用初步(含一个VC2022控制台实例)

前言

有一段时间没有使用libtorch了，最近因为项目的原因，又要拿出来用。

一、Libtorch的下载（cpu版本）

从这个链接开始的下载：

https://pytorch.org/get-started/locally/

打算先下载一个cpu版本的试用一下：

我将2个版本都下载了：

二、创建VisualStudio2022的工程并进行路径设置

采用VS2022，创建一个C++的Console工程，代码如下：

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

int main()
{
    // 创建一个2x3的tensor，初始化为全1
    torch::Tensor tensor = torch::ones({ 2, 3 });

    // 打印tensor
    std::cout << "Original Tensor:\n" << tensor << std::endl;

    // 对tensor执行加法操作
    torch::Tensor add_result = tensor + torch::ones({ 2, 3 });

    // 打印结果
    std::cout << "After adding ones:\n" << add_result << std::endl;

    return 0;
}

包含目录中的路径设置：

库目录中的路径设置：

考虑到libtorch的cpu版本是release版本的。

所以VisualStudio建立的工程也是release x64版本，如下图：

三、编译错误的排除

C++标准由C++14修改为C++17，如下图：

重新编译后前面的错误就消失了。

四、链接错误的排除

出现一系列的链接错误：

修改附加依赖项：

这样一通操作，重新编译后，错误就消失了，如下图：

五、运行错误的解决

将文件所在的目录，放入到系统的path目录当中，如下图：

路径设置好后，

关闭Visual Studio2022，然后重新打开，就可以运行程序了，如下图：

相关源码的下载地址：一个最简单的libtorch C++访问实例