NVIDIA / cuda-python

CUDA Python Low-level Bindings
https://nvidia.github.io/cuda-python/
Other
809 stars 63 forks source link

Support CUDA wheels? #62

Open leofang opened 2 months ago

leofang commented 2 months ago

Currently, CUDA Python loads NVRTC using standard system paths, for example on Linux (cf man ld.so) we do: https://github.com/NVIDIA/cuda-python/blob/2be0aac1a5cac84fec4137aa9c50525653425352/cuda/_cuda/cnvrtc.pyx.in#L56 However, it means unless users purposely add site-packages to LD_LIBRARY_PATH, which is a frown-upon practice, the nvidia-cuda-nvrtc-cuXX wheel (ex: cu12) installed in the user environment will not be used. This issue is for tracking the status of this enhancement.