From 9c05b27ba0619f28ee15b42c309500b6f0958cc1 Mon Sep 17 00:00:00 2001 From: cudawarped <12133430+cudawarped@users.noreply.github.com> Date: Tue, 4 Jun 2024 18:50:06 +0300 Subject: [PATCH] cuda: Add missing python CUDA dependency when CUDA is a first class language --- modules/python/python_loader.cmake | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/modules/python/python_loader.cmake b/modules/python/python_loader.cmake index 4226b82736..41dce57c18 100644 --- a/modules/python/python_loader.cmake +++ b/modules/python/python_loader.cmake @@ -68,8 +68,14 @@ if(DEFINED OPENCV_PYTHON_INSTALL_PATH) endif() set(CMAKE_PYTHON_BINARIES_PATH "${CMAKE_PYTHON_BINARIES_INSTALL_PATH}") if (WIN32 AND HAVE_CUDA) - if (DEFINED CUDA_TOOLKIT_ROOT_DIR) - list(APPEND CMAKE_PYTHON_BINARIES_PATH "os.path.join(os.getenv('CUDA_PATH', '${CUDA_TOOLKIT_ROOT_DIR}'), 'bin')") + if (ENABLE_CUDA_FIRST_CLASS_LANGUAGE) + if (DEFINED CUDAToolkit_LIBRARY_ROOT) + list(APPEND CMAKE_PYTHON_BINARIES_PATH "os.path.join(os.getenv('CUDA_PATH', '${CUDAToolkit_LIBRARY_ROOT}'), 'bin')") + endif() + else() + if (DEFINED CUDA_TOOLKIT_ROOT_DIR) + list(APPEND CMAKE_PYTHON_BINARIES_PATH "os.path.join(os.getenv('CUDA_PATH', '${CUDA_TOOLKIT_ROOT_DIR}'), 'bin')") + endif() endif() endif() string(REPLACE ";" ",\n " CMAKE_PYTHON_BINARIES_PATH "${CMAKE_PYTHON_BINARIES_PATH}")