Skip to content

[BUG] numba-cuda does not support pure CTK wheel users #277

@leofang

Description

@leofang

#155 was only half-done for numba-cuda's wheel support story. During an offline meeting, we agreed to add optional dependencies so that users can do, e.g.

$ pip install numba-cuda[cu12]

to get a self-consistent CUDA 12 environment set up. This was not done.

This issue is compounded by the custom steps in setting up the wheel CI:

# TODO: Replace with rapids-dependency-file-generator
python -m pip install \
psutil \
cffi \
"cuda-python==${CUDA_VER_MAJOR_MINOR}.*" \
"nvidia-cuda-runtime-cu12==${CUDA_VER_MAJOR_MINOR}.*" \
"nvidia-curand-cu12==${CURAND_VER}.*" \
"nvidia-cuda-nvcc-cu12==${CUDA_VER_MAJOR_MINOR}.*" \
"nvidia-cuda-nvrtc-cu12==${CUDA_VER_MAJOR_MINOR}.*" \

which is not documented in the installation guide and does not mimic end user experience.

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions