机器学习笔记 - 在QT/PyTorch/C++ 中加载 TORCHSCRIPT 模型

坐望云起

于 2022-09-30 11:28:24 发布

阅读量1.9k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：深度学习从入门到精通文章标签：机器学习 pytorch c++ libtorch cuda

本文链接：https://blog.youkuaiyun.com/bashendixie5/article/details/126865446

深度学习从入门到精通专栏收录该内容

797 篇文章 ¥99.90 ¥299.90

订阅专栏

本文介绍了如何配置QT+libtorch环境，将PyTorch模型转换为Torch脚本，并在C++中加载执行。在配置过程中详细说明了文件放置和CMakeLists.txt的修改，同时展示了模型对猫咪和赛车的分类测试，验证了模型的正确性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、配置QT+libtorch环境

1、首先下载pytorch，注意版本要和你的python下的pytorch版本，否则运行脚本时会报错，我这里用的是windows下的cuda11.6，release版本，有debug和release两个版本可以下载。

2、将压缩包内的include和lib文件夹放到指定位置，我这里是放到了项目文件夹内。

3、因为QT项目用的是CMakeLists.txt文件，所以添加如下代码。

//两个包含路径
include_directories(${CMAKE_SOURCE_DIR}/lib/torch/include)
include_directories(${CMAKE_SOURCE_DIR}/lib/torch/include/torch/csrc/api/include)

//这里可以把所有的包都引入，我比较懒，暂时这里只引了几个
target_link_libraries(QtFFmpegApp2 PRIVATE ${CMAKE_SOURCE_DIR}/lib/torch/lib/asmjit.lib

了解本专栏