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.
This commit is contained in:
Rifky Bujana Bisri 2023-08-25 15:22:00 -07:00 committed by GitHub
parent 65bf82ffce
commit 2f11a94dd9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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. # 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}) 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")) 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) if(NOT CUDA_cublasLt_LIBRARY)
message(FATAL_ERROR "cuBLASLt library not found") message(FATAL_ERROR "cuBLASLt library not found")
endif() endif()