Skip to content

T
tic
Commit b867db0d by Tatsuya Nishiyama Committed by Tianqi Chen
Options

Fix missing cublas libraries (#1281)

parent 76fa3ca4
Showing with 7 additions and 0 deletions
Jenkinsfile
...@@ -88,6 +88,7 @@ stage('Build') { ...@@ -88,6 +88,7 @@ stage('Build') {
mkdir -p build mkdir -p build
cd build cd build
cp ../cmake/config.cmake . cp ../cmake/config.cmake .
echo set\\(USE_CUBLAS ON\\) >> config.cmake
echo set\\(USE_CUDNN ON\\) >> config.cmake echo set\\(USE_CUDNN ON\\) >> config.cmake
echo set\\(USE_CUDA ON\\) >> config.cmake echo set\\(USE_CUDA ON\\) >> config.cmake
echo set\\(USE_OPENGL ON\\) >> config.cmake echo set\\(USE_OPENGL ON\\) >> config.cmake
......
cmake/util/FindCUDA.cmake
...@@ -50,6 +50,9 @@ macro(find_cuda use_cuda) ...@@ -50,6 +50,9 @@ macro(find_cuda use_cuda)
find_library(CUDA_CUDNN_LIBRARY cudnn find_library(CUDA_CUDNN_LIBRARY cudnn
${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64
${CUDA_TOOLKIT_ROOT_DIR}/lib/win32) ${CUDA_TOOLKIT_ROOT_DIR}/lib/win32)
find_library(CUDA_CUBLAS_LIBRARY cublas
${CUDA_TOOLKIT_ROOT_DIR}/lib/x64
${CUDA_TOOLKIT_ROOT_DIR}/lib/win32)
else(MSVC) else(MSVC)
find_library(_CUDA_CUDA_LIBRARY cuda find_library(_CUDA_CUDA_LIBRARY cuda
PATHS ${CUDA_TOOLKIT_ROOT_DIR} PATHS ${CUDA_TOOLKIT_ROOT_DIR}
...@@ -63,6 +66,9 @@ macro(find_cuda use_cuda) ...@@ -63,6 +66,9 @@ macro(find_cuda use_cuda)
find_library(CUDA_CUDNN_LIBRARY cudnn find_library(CUDA_CUDNN_LIBRARY cudnn
${CUDA_TOOLKIT_ROOT_DIR}/lib64 ${CUDA_TOOLKIT_ROOT_DIR}/lib64
${CUDA_TOOLKIT_ROOT_DIR}/lib) ${CUDA_TOOLKIT_ROOT_DIR}/lib)
find_library(CUDA_CUBLAS_LIBRARY cublas
${CUDA_TOOLKIT_ROOT_DIR}/lib64
${CUDA_TOOLKIT_ROOT_DIR}/lib)
endif(MSVC) endif(MSVC)
endif(CUDA_FOUND) endif(CUDA_FOUND)
endmacro(find_cuda) endmacro(find_cuda)
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