cuda-python icon indicating copy to clipboard operation
cuda-python copied to clipboard

Support CUDA wheels?

Open leofang opened this issue 1 year ago • 0 comments

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.

leofang avatar Apr 24 '24 03:04 leofang