Skip to content

cuda_toolkit_check fails on fresh install as cuda-bindings is not declared in dependencies #145

@jayavenkatesh19

Description

@jayavenkatesh19

While testing out #137, I came across an interesting issue.

On a fresh environment with our declared dependencies (cuda-core>=0.6.0), rapids doctor fails with this message:

cuda_toolkit_check failed
  cuda.bindings 12.x or 13.x must be installed

Looking through the cuda-core docs, cuda-bindings is only pulled in through extras (cuda-core[cu12] or cuda-core[cu13]). So while installing the wheel, everything passes through smoothly but the cuda_toolkit_check throws an error on the first run.

I think one way to fix this would be to add cuda-bindings >=12,<14 to the dependencies.yaml file, and let pip decide on the specific version to install (it worked for me when I tested it out on a CUDA 13 system) , but I am unsure if this would lead to any issues. Adding the [cu12] or [cu13] extras isn't really an option as we want rapids-cli to be used on environments with varying cuda versions.

I think the bare pip install cuda-core shouldn't silently produce a broken install when cuda-bindings is an essential dependency but we should find a way to fix this downstream.

Metadata

Metadata

Labels

bugSomething isn't working

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions