Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
- project(example-app)
- find_package(Torch REQUIRED)
- set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}")
- add_executable(example-app example-app.cpp)
- target_link_libraries(example-app "${TORCH_LIBRARIES}")
- set_property(TARGET example-app PROPERTY CXX_STANDARD 14)
- # 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 example-app
- POST_BUILD
- COMMAND ${CMAKE_COMMAND} -E copy_if_different
- ${TORCH_DLLS}
- $<TARGET_FILE_DIR:example-app>)
- endif (MSVC)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement