From 2f11a94dd92a3ed13937399ed9412af6ed8b7b26 Mon Sep 17 00:00:00 2001 From: Rifky Bujana Bisri Date: Fri, 25 Aug 2023 15:22:00 -0700 Subject: [PATCH] Fix error cublasLt not found on Cuda version 11 and up with Ubuntu 18.0+ As referenced on this thread: https://forums.developer.nvidia.com/t/cublas-for-10-1-is-missing/71015, on Ubuntu 18.04, the cublas library is placed in /usr/lib/x86_64-linux-gnu, where the cublasLt also placed inside that folder. --- CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 3c674e68..a13f93f3 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -410,7 +410,7 @@ if(CUDA_FOUND) # version that ships with CUDA 11 so we force the search to occur inside of the cuda toolkit directory. set(CUDA_LIBS ${CUDA_curand_LIBRARY} ${CUDA_cusparse_LIBRARY} ${CUDA_CUBLAS_LIBRARIES}) if ((CUDA_VERSION VERSION_EQUAL "11.0" OR CUDA_VERSION VERSION_GREATER "11.0")) - find_library(CUDA_cublasLt_LIBRARY NAMES cublasLt PATHS ${CUDA_TOOLKIT_ROOT_DIR}/lib64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 NO_DEFAULT_PATH) + find_library(CUDA_cublasLt_LIBRARY NAMES cublasLt PATHS ${CUDA_TOOLKIT_ROOT_DIR}/lib64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 ${CUDA_TOOLKIT_ROOT_DIR} /usr/lib/x86_64-linux-gnu/ /usr/lib/x86_64-linux-gnu/stubs NO_DEFAULT_PATH) if(NOT CUDA_cublasLt_LIBRARY) message(FATAL_ERROR "cuBLASLt library not found") endif()