From ed29b0a6ed6ed2e758747142457e0092e84b4b91 Mon Sep 17 00:00:00 2001 From: bruno Date: Tue, 13 Oct 2020 14:58:38 +0200 Subject: [PATCH] Add RPATH to torchvision cmake tartet for torch libraries --- CMakeLists.txt | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index cd2a4d49c29..81ca559d530 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -59,13 +59,16 @@ if(MSVC) endif() endif() +include(GNUInstallDirs) +include(CMakePackageConfigHelpers) + add_library(${PROJECT_NAME} SHARED ${MODELS_SOURCES} ${OPERATOR_SOURCES} ${IMAGE_SOURCES}) target_link_libraries(${PROJECT_NAME} PRIVATE ${TORCH_LIBRARIES} ${PNG_LIBRARY} ${JPEG_LIBRARIES} Python3::Python) -set_target_properties(${PROJECT_NAME} PROPERTIES EXPORT_NAME TorchVision) +set_target_properties(${PROJECT_NAME} PROPERTIES + EXPORT_NAME TorchVision + INSTALL_RPATH ${TORCH_INSTALL_PREFIX}/lib) include_directories(torchvision/csrc ${JPEG_INCLUDE_DIRS} ${PNG_INCLUDE_DIRS}) -include(GNUInstallDirs) -include(CMakePackageConfigHelpers) set(TORCHVISION_CMAKECONFIG_INSTALL_DIR "share/cmake/TorchVision" CACHE STRING "install path for TorchVisionConfig.cmake")