diff --git a/cmake/TorchVisionConfig.cmake.in b/cmake/TorchVisionConfig.cmake.in index aa042844753..98f4f58d3c0 100644 --- a/cmake/TorchVisionConfig.cmake.in +++ b/cmake/TorchVisionConfig.cmake.in @@ -28,14 +28,16 @@ include("${CMAKE_CURRENT_LIST_DIR}/${PN}Targets.cmake") if(NOT TARGET torch_library) find_package(Torch REQUIRED) endif() -if (@USE_PYTHON@) +if(@USE_PYTHON@) if(NOT TARGET Python3::Python) find_package(Python3 COMPONENTS Development) endif() endif() -set_target_properties(TorchVision::TorchVision PROPERTIES INTERFACE_INCLUDE_DIRECTORIES "${${PN}_INCLUDE_DIR}" INTERFACE_LINK_LIBRARIES "torch;Python3::Python" ) - +set_target_properties(TorchVision::TorchVision PROPERTIES INTERFACE_INCLUDE_DIRECTORIES "${${PN}_INCLUDE_DIR}" INTERFACE_LINK_LIBRARIES "torch") +if(@USE_PYTHON@) + set_property(TARGET TorchVision::TorchVision APPEND PROPERTY INTERFACE_LINK_LIBRARIES "Python3::Python") +endif() if(@WITH_CUDA@) target_compile_definitions(TorchVision::TorchVision INTERFACE WITH_CUDA)