Skip to content

T
tic
Commit 8a90f877 by Sammy Committed by Tianqi Chen
Options

Fixing issue #1395: Undefined symbol: cudaGetDevice (#1396)

parent 4dd41ddb
Showing with 2 additions and 2 deletions
cmake/util/FindCUDA.cmake
...@@ -28,11 +28,11 @@ macro(find_cuda use_cuda) ...@@ -28,11 +28,11 @@ macro(find_cuda use_cuda)
set(CUDA_INCLUDE_DIRS ${CUDA_TOOLKIT_ROOT_DIR}/include) set(CUDA_INCLUDE_DIRS ${CUDA_TOOLKIT_ROOT_DIR}/include)
set(CUDA_FOUND TRUE) set(CUDA_FOUND TRUE)
if(MSVC) if(MSVC)
find_library(CUDA_CUDAT_LIBRARY cudart find_library(CUDA_CUDART_LIBRARY cudart
${CUDA_TOOLKIT_ROOT_DIR}/lib64 ${CUDA_TOOLKIT_ROOT_DIR}/lib64
${CUDA_TOOLKIT_ROOT_DIR}/lib) ${CUDA_TOOLKIT_ROOT_DIR}/lib)
else(MSVC) else(MSVC)
find_library(CUDA_CUDAT_LIBRARY cudart find_library(CUDA_CUDART_LIBRARY cudart
${CUDA_TOOLKIT_ROOT_DIR}/lib64 ${CUDA_TOOLKIT_ROOT_DIR}/lib64
${CUDA_TOOLKIT_ROOT_DIR}/lib) ${CUDA_TOOLKIT_ROOT_DIR}/lib)
endif(MSVC) endif(MSVC)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment