[Build] fix: TE installation failed to find uv-installed cuDNN libraries

[KivenChen]

Description

Fixes #2206

Type of change

• Bug fix (non-breaking change which fixes an issue)

Changes

Add -DPython_SITEARCH to TE's CMake build args, as a supplement to existing "-DPython_INCLUDE_DIR", "-DPython_EXECUTABLE" etc.

Checklist:

• I have read and followed the contributing guidelines
• The functionality is complete
• I have commented my code, particularly in hard-to-understand areas
• I have made corresponding changes to the documentation
• My changes generate no new warnings
• I have added tests that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes

[timmoon10]

Seems like a reasonable fix for #2206. I see that the cuDNN C++ frontend checks Python_SITEARCH here: https://github.com/NVIDIA/cudnn-frontend/blob/80a8e4af4d89d33a2c59d51fcf9fda1c9d368cd4/cmake/cuDNN.cmake#L5

The DCO check is failing, so can you first sign your commit? https://github.com/NVIDIA/TransformerEngine/blob/main/CONTRIBUTING.rst#sign-your-work

[KivenChen]

Seems like a reasonable fix for #2206. I see that the cuDNN C++ frontend checks Python_SITEARCH here: NVIDIA/cudnn-frontend@80a8e4a/cmake/cuDNN.cmake#L5

The DCO check is failing, so can you first sign your commit? main/CONTRIBUTING.rst#sign-your-work

Hi @timmoon10, the DCO check is now passing and the PR is ready.

[ksivaman]

LGTM

[ksivaman]

/te-ci