diff --git a/CHANGELOG.md b/CHANGELOG.md index e6eaa5cba3..5a2b058d66 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -40,6 +40,7 @@ - PR #2871: Add timing function to utils - PR #2863: in FIL, rename leaf_value_t enums to more descriptive - PR #2892 Update ci/local/README.md +- PR #2910: Adding Support for CuPy 8.x ## Bug Fixes - PR #2882: Allow import on machines without GPUs @@ -64,7 +65,7 @@ - PR #2856: Fix LabelEncoder for filtered input - PR #2855: Updates for RMM being header only - PR #2880: Fix bugs in Auto-ARIMA when s==None -- PR #2877: TSNE exception for n_components > 2 +- PR #2877: TSNE exception for n_components > 2 - PR #2879: Update unit test for LabelEncoder on filtered input # cuML 0.15.0 (Date TBD) diff --git a/ci/mg/build.sh b/ci/mg/build.sh index 9821d28896..9f742edb52 100644 --- a/ci/mg/build.sh +++ b/ci/mg/build.sh @@ -43,7 +43,7 @@ nvidia-smi logger "Activate conda env..." source activate gdf conda install -c conda-forge -c rapidsai -c rapidsai-nightly -c nvidia \ - "cupy>=7,<8.0.0a0" \ + "cupy>7.1.0,<9.0.0a0" \ "cudatoolkit=${CUDA_REL}" \ "cudf=${MINOR_VERSION}" \ "rmm=${MINOR_VERSION}" \ diff --git a/conda/recipes/cuml/meta.yaml b/conda/recipes/cuml/meta.yaml index e62aeecba4..65dddeaffb 100644 --- a/conda/recipes/cuml/meta.yaml +++ b/conda/recipes/cuml/meta.yaml @@ -41,7 +41,7 @@ requirements: - libcuml={{ version }} - libcumlprims {{ minor_version }} - treelite=0.92 - - cupy>=7,<=8.0.0dev.rapidsai0.15 + - cupy>7.1.0,<9.0.0a0 - nccl>=2.5 - ucx-py {{ minor_version }} - dask>=2.12.0 diff --git a/python/README.md b/python/README.md index 1a2faad079..033913c62f 100644 --- a/python/README.md +++ b/python/README.md @@ -4,9 +4,11 @@ This folder contains the Python and Cython code of the algorithms and ML primiti Contents: -- [Build Configuration](#build-configuration) -- [RAFT Integration in cuml.raft](#raft-integration-in-cumlraft) -- [Running Unit Tests](#build-requirements) +- [cuML Python Package](#cuml-python-package) + - [Build Configuration](#build-configuration) + - [RAFT Integration in cuml.raft](#raft-integration-in-cumlraft) + - [Build Requirements](#build-requirements) + - [Python Tests](#python-tests) ### Build Configuration @@ -60,7 +62,7 @@ To build cuML's Python package, the following dependencies are required: - cudf version matching the cuML version - libcuml version matching the cuML version - libcuml={{ version }} -- cupy>=7,<8.0.0a0 +- cupy>7.1.0,<9.0.0a0 - joblib >=0.11 Packages required for multigpu algorithms*: diff --git a/python/cuml/test/test_preprocessing.py b/python/cuml/test/test_preprocessing.py index c478bac62b..0535bf6546 100644 --- a/python/cuml/test/test_preprocessing.py +++ b/python/cuml/test/test_preprocessing.py @@ -574,7 +574,7 @@ def test_kbinsdiscretizer(blobs_dataset, n_bins, # noqa: F811 sk_r_X = transformer.inverse_transform(sk_t_X) if strategy == 'kmeans': - assert_allclose(t_X, sk_t_X, ratio_tol=0.1) + assert_allclose(t_X, sk_t_X, ratio_tol=0.2) else: assert_allclose(t_X, sk_t_X) assert_allclose(r_X, sk_r_X)