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