Qt 使用 LibTorch_qt cxx11abi

Qt 使用 LibTorch

qmake

官网下载 LibTorch 预编译版本

修改 .pro

CONFIG += c++14

QMAKE_CXXFLAGS += -D_GLIBCXX_USE_CXX11_ABI=0
CONFIG += no_keywords

INCLUDEPATH += \
    <path-to-libtorch-folder>/libtorch/include \
    <path-to-libtorch-folder>/libtorch/include/torch/csrc/api/include

LIBS += \
    -L<path-to-libtorch-folder>/libtorch/lib \
	-ltorch \
    -ltorch_cpu \
    -lc10 \
    -lc10_cuda \
    -lcaffe2_nvrtc \
    -lcpuinfo \
    -lprotobuf \
    -lprotobuf-lite \
    -lprotoc \
    -lcaffe2_module_test_dynamic \
    -lcaffe2_detectron_ops_gpu
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

修改 include 方式

#undef slots
#include <torch/torch.h>
#define slots Q_SLOTS
1
2
3

若没有

#include <QObject>
1

可以不作修改

cmake

官网下载LibTorch 预编译版本

创建 Qt 项目

Non-Qt Project
1

CMake
1

修改 CmakeLists.txt

cmake_minimum_required(VERSION 3.5)

project(<project_name> LANGUAGES CXX)

set(CMAKE_CXX_STANDARD 14)
set(CMAKE_CXX_STANDARD_REQUIRED ON)

set(Torch_DIR <path-to-libtorch-folder>/libtorch/share/cmake/Torch)
find_package(Torch REQUIRED)

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

target_link_libraries(${PROJECT_NAME} "${TORCH_LIBRARIES}")
set_property(TARGET ${PROJECT_NAME} PROPERTY CXX_STANDARD 14)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

测试代码

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

using namespace std;

int main()
{
    torch::Tensor tensor = torch::eye(3);
    cout << tensor << endl;
}
1
2
3
4
5
6
7
8
9
10

 1  0  0
 0  1  0
 0  0  1
[ CPUFloatType{3,3} ]
按 <RETURN> 来关闭窗口...
1
2
3
4
5

include 头文件选择

若仅需要推理，则只需要这一个头文件

#include <torch/script.h>
1

尽情享用吧~