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>
pull/2854/head
Prashanth 1 year ago committed by GitHub
parent ba66152ad9
commit 9a9b4af6c5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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)

Loading…
Cancel
Save