mirror of
https://github.com/davisking/dlib.git
synced 2024-11-01 10:14:53 +08:00
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:
parent
ba66152ad9
commit
9a9b4af6c5
@ -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…
Reference in New Issue
Block a user