updated cudnn path (#2843)

* updated cudnn path

* adding cudnn path

* Use same format as command below

---------

Co-authored-by: Davis E. King <davis685@gmail.com>
This commit is contained in:
Prashanth 2023-08-05 13:39:56 +00:00 committed by GitHub
parent ba66152ad9
commit 9a9b4af6c5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -4,7 +4,7 @@ message(STATUS "Looking for cuDNN install...")
# libraries and also a few other places as well.
find_path(cudnn_include cudnn.h
HINTS ${CUDA_INCLUDE_DIRS} ENV CUDNN_INCLUDE_DIR ENV CUDNN_HOME
PATHS /usr/local ENV CPATH
PATHS /usr/local /usr/local/cuda ENV CPATH
PATH_SUFFIXES include
)
get_filename_component(cudnn_hint_path "${CUDA_CUBLAS_LIBRARIES}" PATH)