--- a/cmake/Dependencies.cmake
+++ b/cmake/Dependencies.cmake
@@ -313,12 +313,7 @@
 # --- [ PocketFFT
 set(AT_POCKETFFT_ENABLED 0)
 if(NOT AT_MKL_ENABLED)
-  set(POCKETFFT_INCLUDE_DIR "${Torch_SOURCE_DIR}/third_party/pocketfft/")
-  if(NOT EXISTS "${POCKETFFT_INCLUDE_DIR}")
-    message(FATAL_ERROR "pocketfft directory not found, expected ${POCKETFFT_INCLUDE_DIR}")
-  elseif(NOT EXISTS "${POCKETFFT_INCLUDE_DIR}/pocketfft_hdronly.h")
-    message(FATAL_ERROR "pocketfft headers not found in ${POCKETFFT_INCLUDE_DIR}")
-  endif()
+  find_path(POCKETFFT_INCLUDE_DIR NAMES pocketfft_hdronly.h REQUIRED)
 
   set(AT_POCKETFFT_ENABLED 1)
   message(STATUS "Using pocketfft in directory: ${POCKETFFT_INCLUDE_DIR}")