diff --git a/conda/recipes/rapids-build-env/meta.yaml b/conda/recipes/rapids-build-env/meta.yaml index 58e97dfe..ce39fb7f 100644 --- a/conda/recipes/rapids-build-env/meta.yaml +++ b/conda/recipes/rapids-build-env/meta.yaml @@ -150,8 +150,8 @@ requirements: - treelite {{ treelite_version }} - twine - typing_extensions - - ucx-proc=*=gpu - - ucx + - conda-forge::ucx-proc=*=gpu + - conda-forge::ucx {{ ucx_version }} - umap-learn about: diff --git a/conda/recipes/rapids/meta.yaml b/conda/recipes/rapids/meta.yaml index 68db24f9..833c5df2 100644 --- a/conda/recipes/rapids/meta.yaml +++ b/conda/recipes/rapids/meta.yaml @@ -60,6 +60,8 @@ requirements: - pylibcugraph ={{ minor_version }}.* - libcugraph_etl ={{ minor_version }}.* - ptxcompiler # [linux64] # CUDA enhanced compat. See https://github.com/rapidsai/ptxcompiler + - conda-forge::ucx-proc=*=gpu + - conda-forge::ucx {{ ucx_version }} test: # [linux64] imports: # [linux64] diff --git a/conda/recipes/versions.yaml b/conda/recipes/versions.yaml index 0ce6e7c0..659059cb 100644 --- a/conda/recipes/versions.yaml +++ b/conda/recipes/versions.yaml @@ -167,3 +167,5 @@ treelite_version: - '=2.3.0' transformers_version: - '<=4.10.3' +ucx_version: + - '>=1.12.1'