Clion搭建Libtorch开发项目

最新推荐文章于 2024-12-27 11:05:37 发布

原创

最新推荐文章于 2024-12-27 11:05:37 发布 · 683 阅读

11 ·

CC 4.0 BY-SA版权

文章标签：

#经验分享

本文详细介绍了如何在Clion环境中设置和配置Libtorch开发项目，包括下载libtorch、编写CMake文件、处理动态链接库以及注意事项，特别强调了CMake中不应在debug模式下编写。

Clion搭建Libtorch开发项目

开发环境配置：

1、下载libtorch

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

2、写cmake

cmake_minimum_required(VERSION 3.26 FATAL_ERROR)
project(SSSSSSSS)
# seting libtorch CPU
SET(CMAKE_PREFIX_PATH D:/Libs/libtorch_cpu)
find_package(Torch REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")

add_executable(SSSSSSSS main.cpp )
target_link_libraries(SSSSSSSS "${TORCH_LIBRARIES}")
set_property(TARGET SSSSSSSS PROPERTY CXX_STANDARD 17)

# The following code block is suggested to be used on Windows.
# According to https://github.com/pytorch/pytorch/issues/25457,
# the DLLs need to be copied to avoid memory errors.
if (MSVC)
    file(GLOB TORCH_DLLS "${TORCH_INSTALL_PREFIX}/lib/*.dll")
    add_custom_command(TARGET SSSSSSSS
            POST_BUILD
            COMMAND ${CMAKE_COMMAND} -E copy_if_different
            ${TORCH_DLLS}
            $<TARGET_FILE_DIR:SSSSSSSS>)
endif (MSVC)

3、配置clion的toolchain