11 lines
587 B
Diff
11 lines
587 B
Diff
--- 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()
|
|
|