diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml index af7e9806aa..1dfe0b838c 100644 --- a/.github/workflows/wheels.yml +++ b/.github/workflows/wheels.yml @@ -53,10 +53,10 @@ jobs: skbuild-configure-options: "-DCUML_BUILD_WHEELS=ON -DDETECT_CONDA_ENV=OFF -DCPM_cumlprims_mg_SOURCE=/project/cumlprims_mg/" # Always want to test against latest dask/distributed. - test-before-amd64: "pip install git+https://github.com/dask/dask.git@main git+https://github.com/dask/distributed.git@main git+https://github.com/rapidsai/dask-cuda.git@branch-22.12" + test-before-amd64: "pip install git+https://github.com/dask/dask.git@2022.11.1 git+https://github.com/dask/distributed.git@2022.11.1 git+https://github.com/rapidsai/dask-cuda.git@branch-22.12" # On arm also need to install cupy from the specific webpage and CMake # because treelite needs to be compiled (no wheels available for arm). - test-before-arm64: "pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64 && pip install cmake && pip install git+https://github.com/dask/dask.git@main git+https://github.com/dask/distributed.git@main git+https://github.com/rapidsai/dask-cuda.git@branch-22.12" + test-before-arm64: "pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64 && pip install cmake && pip install git+https://github.com/dask/dask.git@2022.11.1 git+https://github.com/dask/distributed.git@2022.11.1 git+https://github.com/rapidsai/dask-cuda.git@branch-22.12" test-extras: test test-unittest: "pytest -v ./python/cuml/tests -k 'not test_silhouette_score_batched'" secrets: inherit diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh index 4638e6bb14..cacef5a87b 100644 --- a/ci/gpu/build.sh +++ b/ci/gpu/build.sh @@ -39,10 +39,10 @@ export UCX_PY_VERSION='0.29.*' export NUMBA_THREADING_LAYER=workqueue # Whether to install dask nightly or stable packages -export INSTALL_DASK_MAIN=1 +export INSTALL_DASK_MAIN=0 # Dask version to install when `INSTALL_DASK_MAIN=0` -export DASK_STABLE_VERSION="2022.9.2" +export DASK_STABLE_VERSION="2022.11.1" ################################################################################ # SETUP - Check environment @@ -107,7 +107,7 @@ function install_dask { conda list else gpuci_logger "gpuci_mamba_retry install conda-forge::dask=={$DASK_STABLE_VERSION} conda-forge::distributed=={$DASK_STABLE_VERSION} conda-forge::dask-core=={$DASK_STABLE_VERSION} --force-reinstall" - gpuci_mamba_retry install conda-forge::dask=={$DASK_STABLE_VERSION} conda-forge::distributed=={$DASK_STABLE_VERSION} conda-forge::dask-core=={$DASK_STABLE_VERSION} --force-reinstall + gpuci_mamba_retry install conda-forge::dask==$DASK_STABLE_VERSION conda-forge::distributed==$DASK_STABLE_VERSION conda-forge::dask-core==$DASK_STABLE_VERSION --force-reinstall fi set +x } diff --git a/conda/environments/cuml_dev_cuda11.0.yml b/conda/environments/cuml_dev_cuda11.0.yml index 530667174f..00ac1fec1f 100644 --- a/conda/environments/cuml_dev_cuda11.0.yml +++ b/conda/environments/cuml_dev_cuda11.0.yml @@ -22,8 +22,8 @@ dependencies: - pylibraft=22.12.* - dask-cudf=22.12.* - dask-cuda=22.12.* -- dask>=2022.9.2 -- distributed>=2022.9.2 +- dask==2022.11.1 +- distributed==2022.11.1 - ucx>=1.13.0 - ucx-py=0.29.* - ucx-proc=*=gpu diff --git a/conda/environments/cuml_dev_cuda11.2.yml b/conda/environments/cuml_dev_cuda11.2.yml index acbdc127b1..768de74699 100644 --- a/conda/environments/cuml_dev_cuda11.2.yml +++ b/conda/environments/cuml_dev_cuda11.2.yml @@ -22,8 +22,8 @@ dependencies: - pylibraft=22.12.* - dask-cudf=22.12.* - dask-cuda=22.12.* -- dask>=2022.9.2 -- distributed>=2022.9.2 +- dask==2022.11.1 +- distributed==2022.11.1 - ucx>=1.13.0 - ucx-py=0.29.* - ucx-proc=*=gpu diff --git a/conda/environments/cuml_dev_cuda11.4.yml b/conda/environments/cuml_dev_cuda11.4.yml index bd17a5f461..61334696e1 100644 --- a/conda/environments/cuml_dev_cuda11.4.yml +++ b/conda/environments/cuml_dev_cuda11.4.yml @@ -22,8 +22,8 @@ dependencies: - pylibraft=22.12.* - dask-cudf=22.12.* - dask-cuda=22.12.* -- dask>=2022.9.2 -- distributed>=2022.9.2 +- dask==2022.11.1 +- distributed==2022.11.1 - ucx>=1.13.0 - ucx-py=0.29.* - ucx-proc=*=gpu diff --git a/conda/environments/cuml_dev_cuda11.5.yml b/conda/environments/cuml_dev_cuda11.5.yml index e240fcb16e..fcac29e751 100644 --- a/conda/environments/cuml_dev_cuda11.5.yml +++ b/conda/environments/cuml_dev_cuda11.5.yml @@ -22,8 +22,8 @@ dependencies: - pylibraft=22.12.* - dask-cudf=22.12.* - dask-cuda=22.12.* -- dask>=2022.9.2 -- distributed>=2022.9.2 +- dask==2022.11.1 +- distributed==2022.11.1 - ucx>=1.13.0 - ucx-py=0.29.* - ucx-proc=*=gpu diff --git a/conda/recipes/cuml/meta.yaml b/conda/recipes/cuml/meta.yaml index 87e6b79ee1..c3d5e77ae0 100644 --- a/conda/recipes/cuml/meta.yaml +++ b/conda/recipes/cuml/meta.yaml @@ -59,8 +59,8 @@ requirements: - nccl>=2.9.9 - ucx-py {{ ucx_py_version }} - ucx-proc=*=gpu - - dask>=2022.9.2 - - distributed>=2022.9.2 + - dask==2022.11.1 + - distributed==2022.11.1 - joblib >=0.11 - {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }} - cuda-python >=11.7.1,<12.0