blob: 77905dbd1ac857092b207db38c8c9a48f5055125 (
plain)
1
2
3
4
5
6
7
8
9
10
11
|
--- a/cmake/Dependencies.cmake 2024-04-29 18:37:34.005639858 +0200
+++ b/cmake/Dependencies.cmake 2024-04-29 18:39:29.126587738 +0200
@@ -1235,7 +1235,7 @@
if(CUDNN_VERSION VERSION_LESS 8.5)
message(FATAL_ERROR "PyTorch needs CuDNN-8.5 or above, but found ${CUDNN_VERSION}. Builds are still possible with `USE_CUDNN=0`")
endif()
- set(CUDNN_FRONTEND_INCLUDE_DIR ${CMAKE_CURRENT_LIST_DIR}/../third_party/cudnn_frontend/include)
+ set(CUDNN_FRONTEND_INCLUDE_DIR /opt/cuda/include)
target_include_directories(torch::cudnn INTERFACE ${CUDNN_FRONTEND_INCLUDE_DIR})
endif()
|
