mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

dapted to cudnn cudnn 8.6

This commit is contained in:
qiuzhongya 2022-11-07 17:19:31 +08:00
parent 7262ee8ac9
commit ef814a261a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/CMakeLists.txt
View File

@ -119,7 +119,7 @@ if(ENABLE_GPU AND GPU_BACKEND_CUDA)
if(DEFINED ENV{CUDNN_HOME} AND NOT $ENV{CUDNN_HOME} STREQUAL "")
set(CUDNN_INCLUDE_DIR $ENV{CUDNN_HOME}/include)
if(WIN32)
set(CUDNN_LIBRARY_DIR $ENV{CUDNN_HOME}/lib)
set(CUDNN_LIBRARY_DIR $ENV{CUDNN_HOME}/lib $ENV{CUDNN_HOME}/lib/x64)
else()
set(CUDNN_LIBRARY_DIR $ENV{CUDNN_HOME}/lib64)
endif()