366f3cb9d8
* Ensure RMM is 0.18 or later * Add use_rmm flag to global configuration * Modify XGBCachingDeviceAllocatorImpl to skip CUB when use_rmm=True * Update the demo * [CI] Pin NumPy to 1.19.4, since NumPy 1.19.5 doesn't work with latest Shap
41 lines
513 B
YAML
41 lines
513 B
YAML
name: cpu_test
|
|
channels:
|
|
- conda-forge
|
|
dependencies:
|
|
- python=3.7
|
|
- pip
|
|
- wheel
|
|
- pyyaml
|
|
- cpplint
|
|
- pylint
|
|
- numpy=1.19.4
|
|
- scipy
|
|
- scikit-learn
|
|
- pandas
|
|
- matplotlib
|
|
- dask
|
|
- distributed
|
|
- python-graphviz
|
|
- hypothesis
|
|
- astroid
|
|
- sphinx
|
|
- sh
|
|
- recommonmark
|
|
- mock
|
|
- breathe
|
|
- pytest
|
|
- pytest-cov
|
|
- python-kubernetes
|
|
- urllib3
|
|
- jsonschema
|
|
- boto3
|
|
- awscli
|
|
- numba
|
|
- llvmlite
|
|
- pip:
|
|
- shap
|
|
- ipython # required by shap at import time.
|
|
- guzzle_sphinx_theme
|
|
- datatable
|
|
- modin[all]
|