cuda-python
cuda-python copied to clipboard
[FEA]: Support `load_nvidia_dynamic_lib()` `"cuda"`, `"nvml"`
xref: https://github.com/NVIDIA/cuda-python/pull/1284#discussion_r2558243942
These libs are installed as part of the CUDA driver and can be reliably found via the dynamic loader system search.
-
"cuda"- Linux:
libcuda.so.1 - Windows:
nvcuda.dll
- Linux:
-
"nvml"- Linux:
libnvidia-ml.so.1 - Windows:
nvml.dll
- Linux:
The "cuda" lib can come from cuda-compat (Linux only) but it must be on the system path, so we can simply rely on the dynamic loader and do not need to search in multiple places.