diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml index fcbf1451a8..cb9edd45e5 100644 --- a/conda/environments/all_cuda-118_arch-x86_64.yaml +++ b/conda/environments/all_cuda-118_arch-x86_64.yaml @@ -22,6 +22,7 @@ dependencies: - dask-core==2023.9.2 - dask-cuda==23.12.* - dask==2023.9.2 +- distributed-ucxx==0.35.* - distributed==2023.9.2 - doxygen>=1.8.20 - gcc_linux-64=11.* diff --git a/conda/environments/all_cuda-120_arch-x86_64.yaml b/conda/environments/all_cuda-120_arch-x86_64.yaml index a1e22f50a0..e117e27ffe 100644 --- a/conda/environments/all_cuda-120_arch-x86_64.yaml +++ b/conda/environments/all_cuda-120_arch-x86_64.yaml @@ -23,6 +23,7 @@ dependencies: - dask-core==2023.9.2 - dask-cuda==23.12.* - dask==2023.9.2 +- distributed-ucxx==0.35.* - distributed==2023.9.2 - doxygen>=1.8.20 - gcc_linux-64=11.* diff --git a/dependencies.yaml b/dependencies.yaml index 9baacdab4f..e715376e63 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -444,6 +444,7 @@ dependencies: - ucx>=1.13.0 - ucx-proc=*=gpu - &ucx_py_conda ucx-py==0.35.* + - distributed-ucxx==0.35.* - output_types: pyproject packages: - &pylibraft_conda pylibraft==23.12.*