diff --git a/cmake/torch.cmake b/cmake/torch.cmake index 6c5829d..7437c5b 100644 --- a/cmake/torch.cmake +++ b/cmake/torch.cmake @@ -8,6 +8,7 @@ execute_process( OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_VARIABLE TORCH_DIR ) +message(STATUS "TORCH_DIR: ${TORCH_DIR}") list(APPEND CMAKE_PREFIX_PATH "${TORCH_DIR}") find_package(Torch REQUIRED)