Skip to content

[FEA] CUDA DLLs from Torch GPU #256

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
gmarkall opened this issue May 20, 2025 · 5 comments
Open

[FEA] CUDA DLLs from Torch GPU #256

gmarkall opened this issue May 20, 2025 · 5 comments
Labels
feature request New feature or request

Comments

@gmarkall
Copy link
Collaborator

From @fbordignon in numba/numba#10074:

Hi guys, I would like to run an Idea by you.
We are deploying numba within an app that already has torch with packaged cuda dlls

Would you consider merging a change to add this option for CUDA installation?

The idea is to change cuda_paths.py by adding the torch-cuda option and auxiliar functions to find the dlls inside the site-packages folder.

tyia

@gmarkall gmarkall added the feature request New feature or request label May 20, 2025
@gmarkall gmarkall changed the title [FEA] [FEA] CUDA DLLs from Torch GPU May 20, 2025
@gmarkall
Copy link
Collaborator Author

I just realised this should really become a question for the CUDA Python Path Finder rather than Numba-CUDA's implementation - @rwgk what do you think of this suggested change? (Should this issue move to https://github.com/NVIDIA/cuda-python?)

@leofang
Copy link
Member

leofang commented May 20, 2025

We already handle the CUDA wheel lookup/loading that PyTorch and all other projects need, so feel free to close this issue! 🙂
https://github.com/NVIDIA/cuda-python/blob/main/cuda_bindings/cuda/bindings/_path_finder/README.md

@leofang
Copy link
Member

leofang commented May 20, 2025

(For the record there are some loose ends that @rwgk will address to make it useable by end users, the tracking epic is here: NVIDIA/cuda-python#451)

@gmarkall
Copy link
Collaborator Author

Thanks @leofang - Numba-CUDA still needs to make use of the path finder, so I'll leave this open until that's done.

@leofang
Copy link
Member

leofang commented May 20, 2025

Numba-CUDA still needs to make use of the path finder

I think this is already tracked once we complete #128 (and its child issues).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants