From eb0f794fc6b91fef7797695a0e9928f6e213b111 Mon Sep 17 00:00:00 2001 From: John Kirkham Date: Fri, 28 May 2021 10:36:26 -0700 Subject: [PATCH] Use UCX-Py 0.20 --- conda/environments/cuml_dev_cuda10.1.yml | 2 +- conda/environments/cuml_dev_cuda10.2.yml | 2 +- conda/environments/cuml_dev_cuda11.0.yml | 2 +- conda/environments/cuml_dev_cuda11.2.yml | 2 +- conda/recipes/cuml/meta.yaml | 4 ++-- conda/recipes/libcuml/meta.yaml | 4 ++-- 6 files changed, 8 insertions(+), 8 deletions(-) diff --git a/conda/environments/cuml_dev_cuda10.1.yml b/conda/environments/cuml_dev_cuda10.1.yml index 2726ccc84e..e59392b43a 100644 --- a/conda/environments/cuml_dev_cuda10.1.yml +++ b/conda/environments/cuml_dev_cuda10.1.yml @@ -14,7 +14,7 @@ dependencies: - libcumlprims=21.06.* - dask-cudf=21.06.* - dask-cuda=21.06.* -- ucx-py=21.06.* +- ucx-py=0.20.* - ucx-proc=*=gpu - dask-ml - doxygen>=1.8.20 diff --git a/conda/environments/cuml_dev_cuda10.2.yml b/conda/environments/cuml_dev_cuda10.2.yml index d1c5163bf8..b0003188ec 100644 --- a/conda/environments/cuml_dev_cuda10.2.yml +++ b/conda/environments/cuml_dev_cuda10.2.yml @@ -14,7 +14,7 @@ dependencies: - libcumlprims=21.06.* - dask-cudf=21.06.* - dask-cuda=21.06.* -- ucx-py=21.06.* +- ucx-py=0.20.* - ucx-proc=*=gpu - dask-ml - doxygen>=1.8.20 diff --git a/conda/environments/cuml_dev_cuda11.0.yml b/conda/environments/cuml_dev_cuda11.0.yml index 2326ff755b..ae8ef3a836 100644 --- a/conda/environments/cuml_dev_cuda11.0.yml +++ b/conda/environments/cuml_dev_cuda11.0.yml @@ -14,7 +14,7 @@ dependencies: - libcumlprims=21.06.* - dask-cudf=21.06.* - dask-cuda=21.06.* -- ucx-py=21.06.* +- ucx-py=0.20.* - ucx-proc=*=gpu - dask-ml - doxygen>=1.8.20 diff --git a/conda/environments/cuml_dev_cuda11.2.yml b/conda/environments/cuml_dev_cuda11.2.yml index 9a2ce45b8a..02f93feb33 100644 --- a/conda/environments/cuml_dev_cuda11.2.yml +++ b/conda/environments/cuml_dev_cuda11.2.yml @@ -14,7 +14,7 @@ dependencies: - libcumlprims=21.06.* - dask-cudf=21.06.* - dask-cuda=21.06.* -- ucx-py=21.06.* +- ucx-py=0.20.* - ucx-proc=*=gpu - dask-ml - doxygen>=1.8.20 diff --git a/conda/recipes/cuml/meta.yaml b/conda/recipes/cuml/meta.yaml index 03961f43ea..f685014ebb 100644 --- a/conda/recipes/cuml/meta.yaml +++ b/conda/recipes/cuml/meta.yaml @@ -33,7 +33,7 @@ requirements: - libcuml={{ version }} - libcumlprims {{ minor_version }} - cudatoolkit {{ cuda_version }}.* - - ucx-py {{ minor_version }} + - ucx-py 0.20 - ucx-proc=*=gpu run: - python x.x @@ -44,7 +44,7 @@ requirements: - cupy>=7.8.0,<10.0.0a0 - treelite=1.3.0 - nccl>=2.8.4 - - ucx-py {{ minor_version }} + - ucx-py 0.20 - ucx-proc=*=gpu - dask>=2.12.0 - distributed>=2.12.0 diff --git a/conda/recipes/libcuml/meta.yaml b/conda/recipes/libcuml/meta.yaml index 010636aa6c..920a171e97 100644 --- a/conda/recipes/libcuml/meta.yaml +++ b/conda/recipes/libcuml/meta.yaml @@ -41,7 +41,7 @@ requirements: - nccl >=2.8.4 - cudf {{ minor_version }} - cudatoolkit {{ cuda_version }}.* - - ucx-py {{ minor_version }} + - ucx-py 0.20 - ucx-proc=*=gpu - libcumlprims {{ minor_version }} - lapack @@ -54,7 +54,7 @@ requirements: - libcumlprims {{ minor_version }} - cudf {{ minor_version }} - nccl>=2.8.4 - - ucx-py {{ minor_version }} + - ucx-py 0.20 - ucx-proc=*=gpu - {{ pin_compatible('cudatoolkit', max_pin='x.x') }} - treelite=1.3.0