Backport PR #837: Add path for CUDA static library based on CUDA_HOME

[Copilot]

This PR backports the changes from PR #837 to the 12.9.x branch to improve the installation process by automatically detecting CUDA library paths based on CUDA_HOME instead of requiring manual LIBRARY_PATH environment variable setup.

Problem

Previously, both CUDA_HOME (or CUDA_PATH) and LIBRARY_PATH were required to be set in the environment during installation. While there was already a check for CUDA_HOME in setup.py, there was no check for LIBRARY_PATH. When LIBRARY_PATH was not set, the installation would fail at link time with an error indicating that cudart_static could not be found.

Solution

This backport makes two key changes:

1. Remove manual environment variable setup from CI workflow (.github/workflows/build-wheel.yml):

   • Removed LIBRARY_PATH=/host/${{ env.CUDA_PATH }}/lib from Linux environment
   • Removed LIB="${CUDA_HOME}\\lib\\x64;${LIB}" from Windows environment

2. Add automatic CUDA library path detection (cuda_bindings/setup.py):

   • Automatically detects and adds CUDA library directories based on CUDA_HOME
   • For Linux: searches lib64 and lib subdirectories
   • For Windows: searches lib\x64 subdirectory
   • Uses the pattern: library_dirs.extend(os.path.join(prefix, subdir) for prefix in CUDA_HOME for subdir in cudalib_subdirs)

Impact

This change makes the installation process more robust and user-friendly by:

• Eliminating the need for users to manually set LIBRARY_PATH environment variables
• Automatically finding CUDA libraries in standard locations based on CUDA_HOME
• Reducing installation failures due to missing library path configuration

The backport maintains full compatibility with existing installations while improving the experience for new users.

---

✨ Let Copilot coding agent set things up for you — coding agent works faster and does higher quality work when set up for your repo.

[copy-pr-bot]

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

[leofang]

/ok to test ccc8b08