diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh index e646307cf6..e996eb201c 100755 --- a/ci/gpu/build.sh +++ b/ci/gpu/build.sh @@ -122,8 +122,8 @@ if [[ -z "$PROJECT_FLASH" || "$PROJECT_FLASH" == "0" ]]; then gpuci_logger "Install the main version of dask and distributed" set -x - pip install "git+https://github.com/dask/distributed.git@main" --upgrade --no-deps - pip install "git+https://github.com/dask/dask.git@main" --upgrade --no-deps + pip install "git+https://github.com/dask/distributed.git@2021.09.1" --upgrade --no-deps + pip install "git+https://github.com/dask/dask.git@2021.09.1" --upgrade --no-deps set +x gpuci_logger "Python pytest for cuml" @@ -195,8 +195,8 @@ else gpuci_logger "Install the main version of dask and distributed" set -x - pip install "git+https://github.com/dask/distributed.git@main" --upgrade --no-deps - pip install "git+https://github.com/dask/dask.git@main" --upgrade --no-deps + pip install "git+https://github.com/dask/distributed.git@2021.09.1" --upgrade --no-deps + pip install "git+https://github.com/dask/dask.git@2021.09.1" --upgrade --no-deps set +x gpuci_logger "Building cuml" diff --git a/conda/environments/cuml_dev_cuda11.0.yml b/conda/environments/cuml_dev_cuda11.0.yml index 9fdecefa21..21457b507a 100644 --- a/conda/environments/cuml_dev_cuda11.0.yml +++ b/conda/environments/cuml_dev_cuda11.0.yml @@ -30,8 +30,8 @@ dependencies: - pip - pip: - sphinx_markdown_tables - - git+https://github.com/dask/dask.git@main - - git+https://github.com/dask/distributed.git@main + - git+https://github.com/dask/dask.git@2021.09.1 + - git+https://github.com/dask/distributed.git@2021.09.1 # rapids-build-env, notebook-env and doc-env are defined in # https://docs.rapids.ai/maintainers/depmgmt/ diff --git a/conda/environments/cuml_dev_cuda11.2.yml b/conda/environments/cuml_dev_cuda11.2.yml index 7759958426..f2c5a99bf8 100644 --- a/conda/environments/cuml_dev_cuda11.2.yml +++ b/conda/environments/cuml_dev_cuda11.2.yml @@ -30,8 +30,8 @@ dependencies: - pip - pip: - sphinx_markdown_tables - - git+https://github.com/dask/dask.git@main - - git+https://github.com/dask/distributed.git@main + - git+https://github.com/dask/dask.git@2021.09.1 + - git+https://github.com/dask/distributed.git@2021.09.1 # rapids-build-env, notebook-env and doc-env are defined in # https://docs.rapids.ai/maintainers/depmgmt/ diff --git a/conda/environments/cuml_dev_cuda11.4.yml b/conda/environments/cuml_dev_cuda11.4.yml index 57505c9e78..f07a24e0e4 100644 --- a/conda/environments/cuml_dev_cuda11.4.yml +++ b/conda/environments/cuml_dev_cuda11.4.yml @@ -30,8 +30,8 @@ dependencies: - pip - pip: - sphinx_markdown_tables - - git+https://github.com/dask/dask.git@main - - git+https://github.com/dask/distributed.git@main + - git+https://github.com/dask/dask.git@2021.09.1 + - git+https://github.com/dask/distributed.git@2021.09.1 # rapids-build-env, notebook-env and doc-env are defined in # https://docs.rapids.ai/maintainers/depmgmt/ diff --git a/conda/recipes/cuml/meta.yaml b/conda/recipes/cuml/meta.yaml index 1eadec7d64..5e517cd5b6 100644 --- a/conda/recipes/cuml/meta.yaml +++ b/conda/recipes/cuml/meta.yaml @@ -46,8 +46,8 @@ requirements: - nccl>=2.9.9 - ucx-py 0.22 - ucx-proc=*=gpu - - dask>=2021.6.0 - - distributed>=2021.6.0 + - dask=2021.09.1 + - distributed=2021.09.1 - joblib >=0.11 - {{ pin_compatible('cudatoolkit', max_pin='x.x') }}