fix cuda path for some installation

4f19cc12 · jeremad · jeremad · 7cf79a5e · 4f19cc12
Commit 4f19cc12 authored Apr 19, 2019 by jeremad Committed by jeremad Apr 24, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

CUDA.cmake cmake/CUDA.cmake +2 -2

No files found.
--- a/cmake/CUDA.cmake
+++ b/cmake/CUDA.cmake
@@ -37,9 +37,9 @@ if (LINUX)
    find_library(OPENGL_LIBRARIES GL NO_CMAKE_FIND_ROOT_PATH)
  else(TEGRA_DEMO)
    find_package(GLEW)
-    find_library(CUDA cuda)
+    find_library(CUDA cuda PATHS "${LINUX_CUDA_PATH}/stubs")
    find_library(CUDART cudart PATHS ${LINUX_CUDA_PATH} NO_DEFAULT_PATH)
-    find_library(NVML nvidia-ml)
+    find_library(NVML nvidia-ml PATHS "${LINUX_CUDA_PATH}/stubs")
  endif(TEGRA_DEMO)
  include_directories(${CMAKE_EXTERNAL_DEPS}/include)
  link_directories(${LINUX_CUDA_PATH})