diff --git a/dependencies.yaml b/dependencies.yaml index 8f5061dc76..dca6bd0164 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -123,6 +123,7 @@ files: extras: table: project includes: + - cuda_wheels - depends_on_cuda_python - depends_on_cudf - depends_on_cupy diff --git a/python/cuml/pyproject.toml b/python/cuml/pyproject.toml index cf4bb36461..5a96d98874 100644 --- a/python/cuml/pyproject.toml +++ b/python/cuml/pyproject.toml @@ -102,6 +102,11 @@ dependencies = [ "libcuml==25.2.*,>=0.0.0a0", "numba>=0.57", "numpy>=1.23,<3.0a0", + "nvidia-cublas", + "nvidia-cufft", + "nvidia-curand", + "nvidia-cusolver", + "nvidia-cusparse", "packaging", "pylibraft==25.2.*,>=0.0.0a0", "raft-dask==25.2.*,>=0.0.0a0", @@ -171,7 +176,7 @@ versioneer\.py | [tool.rapids-build-backend] build-backend = "scikit_build_core.build" dependencies-file = "../../dependencies.yaml" -matrix-entry = "cuda_suffixed=true" +matrix-entry = "cuda_suffixed=true;use_cuda_wheels=true" requires = [ "cmake>=3.26.4,!=3.30.0",