diff --git a/python/pylibraft/_custom_build/backend.py b/python/pylibraft/_custom_build/backend.py index 209e9e4b67..1fe82860c1 100644 --- a/python/pylibraft/_custom_build/backend.py +++ b/python/pylibraft/_custom_build/backend.py @@ -19,7 +19,7 @@ def replace_requirements(func): def wrapper(config_settings=None): orig_list = getattr(_orig, func.__name__)(config_settings) cuda_suffix = os.getenv("RAPIDS_PY_WHEEL_CUDA_SUFFIX", default="") - append_list = [f"rmm{cuda_suffix}==23.2.*"] + append_list = [f"rmm{cuda_suffix}==23.4.*"] return orig_list + append_list return wrapper diff --git a/python/pylibraft/setup.py b/python/pylibraft/setup.py index 230b9127e3..cc429f87b0 100644 --- a/python/pylibraft/setup.py +++ b/python/pylibraft/setup.py @@ -25,7 +25,7 @@ install_requires = [ "numpy", "cuda-python>=11.7.1,<12.0", - f"rmm{cuda_suffix}==23.2.*", + f"rmm{cuda_suffix}==23.4.*", ] extras_require = { diff --git a/python/raft-dask/setup.py b/python/raft-dask/setup.py index 5feddb626e..e7462c3559 100644 --- a/python/raft-dask/setup.py +++ b/python/raft-dask/setup.py @@ -26,11 +26,11 @@ "numpy", "numba>=0.49", "joblib>=0.11", - "dask-cuda==23.2.*", + "dask-cuda==23.4.*", "dask==2023.1.1", - f"ucx-py{cuda_suffix}==0.30.*", + f"ucx-py{cuda_suffix}==0.31.*", "distributed==2023.1.1", - f"pylibraft{cuda_suffix}==23.2.*", + f"pylibraft{cuda_suffix}==23.4.*", ] extras_require = {