Qt使用libtorch（2023.1）

最近想用qt使用由pytorch生成训练好的神经网络，所以需要先在qt上配置一下libtorch。也是看了各种的教程或者经验贴，但还是捣鼓了半天，最后终于用msvc2017编译器+cmake成功了。本来一开始我是尝试用qmake，在.pro里面直接添加库、路径之类，但这个最后编译会报错，差不多如这篇帖子所说的https://www.cnblogs.com/allentbky/p/14315040.html，但我最后发现不仅有IValue这类文件的错误，还有Operator.h等文件上的错误（估计是因为我的libtorch版本更新的原因），都是一些语法错误，很烦，我也不想去注释源码（感觉也不一定能解决），所以最后还是投奔了cmake，没想到很快啊，就编译跑通了（果然还是官网推荐的方法好用）。下面给还刚接触的朋友梳理一下配置过程。

首先说一下我的各个工具和软件的版本，

Qt的版本是5.12.2，creator的版本是5.0.2，官网的下载地址是https://www.qt.io/offline-installers（自己找有点难），这是官网能免费提供且有exe安装程序的最好版本。安装时候把msvc2017点上，其他的自己看着点（建议把mingw也点上）

安装好后如果你之前没装过visual studio，那么qt的编译器里是不会出现msvc的，kits里面的msvc前面也会有黄色感叹号，我就是这种情况，所以需要再装一下vs2017（最小化安装即可，如下图所示，或按照这篇帖子https://blog.csdn.net/weixin_43712556/article/details/125906511?utm_medium=distribute.pc_relevant.none-task-blog-2~default~baidujs_baidulandingword~default-5-125906511-blog-127731223.pc_relevant_3mothn_strategy_and_data_recovery&spm=1001.2101.3001.4242.4&utm_relevant_index=8）

然后就是装一下win10 sdk，只需要安装debug tools（具体也可以看上面那篇帖子）。安装好这两项后，qt应该就能自己检测出msvc的编译器和win10 kits中的debugger了，如下图所示

最后再装一下cmake，qt也会自动检测到

当然，还有最重要的libtorch，官网就有编译好的，我下的是cpu的版本

到这里所有需要下载配置的工具都齐了，可以创个demo看看能否成功运行。我是参照这篇文章进行cmakelist的改写https://blog.csdn.net/qq_49466306/article/details/110241031，如果你创建的是qt widget程序，注意调整一下几条语句的位置和顺序，我的cmakelist如下

cmake_minimum_required(VERSION 3.5)

project(test2 VERSION 0.1 LANGUAGES CXX)

set(CMAKE_INCLUDE_CURRENT_DIR ON)

set(CMAKE_AUTOUIC ON)
set(CMAKE_AUTOMOC ON)
set(CMAKE_AUTORCC ON)

set(CMAKE_CXX_STANDARD 14)
set(CMAKE_CXX_STANDARD_REQUIRED ON)

find_package(QT NAMES Qt6 Qt5 COMPONENTS Widgets REQUIRED)
find_package(Qt${QT_VERSION_MAJOR} COMPONENTS Widgets REQUIRED)

#改成你自己的libtorch地址
set(Torch_DIR E:/libtorch/share/cmake/Torch)
find_package(Torch REQUIRED)

FILE(GLOB SC_FILES "*.cpp" "*.h")
#add_executable(${PROJECT_NAME} ${SC_FILES})

set(PROJECT_SOURCES
        main.cpp
        mainwindow.cpp
        mainwindow.h
        mainwindow.ui
)

if(${QT_VERSION_MAJOR} GREATER_EQUAL 6)
    qt_add_executable(test2
        MANUAL_FINALIZATION
        ${PROJECT_SOURCES}
    )
# Define target properties for Android with Qt 6 as:
#    set_property(TARGET test2 APPEND PROPERTY QT_ANDROID_PACKAGE_SOURCE_DIR
#                 ${CMAKE_CURRENT_SOURCE_DIR}/android)
# For more information, see https://doc.qt.io/qt-6/qt-add-executable.html#target-creation
else()
    if(ANDROID)
        add_library(test2 SHARED
            ${PROJECT_SOURCES}
        )
# Define properties for Android with Qt 5 after find_package() calls as:
#    set(ANDROID_PACKAGE_SOURCE_DIR "${CMAKE_CURRENT_SOURCE_DIR}/android")
    else()
        add_executable(test2
            ${PROJECT_SOURCES}
        )
    endif()
endif()

#注意这里把原来有的private删掉了
target_link_libraries(test2 Qt${QT_VERSION_MAJOR}::Widgets)

set_target_properties(test2 PROPERTIES
    MACOSX_BUNDLE_GUI_IDENTIFIER my.example.com
    MACOSX_BUNDLE_BUNDLE_VERSION ${PROJECT_VERSION}
    MACOSX_BUNDLE_SHORT_VERSION_STRING ${PROJECT_VERSION_MAJOR}.${PROJECT_VERSION_MINOR}
)
target_link_libraries(${PROJECT_NAME} "${TORCH_LIBRARIES}")

set_property(TARGET ${PROJECT_NAME} PROPERTY CXX_STANDARD 14)

if(QT_VERSION_MAJOR EQUAL 6)
    qt_finalize_executable(test2)
endif()

有个坑就是由于前后调用了两次target_link_libraries，所以要么都得是private，要么都不是，不然会报错

还有就是编译器要记得选成msvc，如下图所示

mainwindow.cpp的代码如下

#include <torch/torch.h>
#include "mainwindow.h"
#include "./ui_mainwindow.h"

using namespace std;

MainWindow::MainWindow(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::MainWindow)
{
    ui->setupUi(this);
    torch::Tensor tensor = torch::eye(3);
        cout << tensor << endl;
}

MainWindow::~MainWindow()
{
    delete ui;
}

如果成功编译运行并打印出如下结果就说明没什么问题了

完整的项目代码可见https://github.com/myccmj/qt-libtorch-cmake