diff --git a/cmake/cudnn.cmake b/cmake/cudnn.cmake index 34cc9c8199b372b51ee3d8219c71431f7f6eb745..04da3db951972286c82d93207a04e9f847105d4e 100644 --- a/cmake/cudnn.cmake +++ b/cmake/cudnn.cmake @@ -29,11 +29,13 @@ list( ${CUDNN_ROOT} ${CUDNN_ROOT}/lib64 ${CUDNN_ROOT}/lib + ${CUDNN_ROOT}/lib/x64 ${CUDNN_ROOT}/lib/${TARGET_ARCH}-linux-gnu ${CUDNN_ROOT}/local/cuda-${CUDA_VERSION}/targets/${TARGET_ARCH}-linux/lib/ $ENV{CUDNN_ROOT} $ENV{CUDNN_ROOT}/lib64 $ENV{CUDNN_ROOT}/lib + $ENV{CUDNN_ROOT}/lib/x64 /usr/lib ${CUDA_TOOLKIT_ROOT_DIR} ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64)