diff --git a/conda/environments/cuml_dev_cuda11.4.yml b/conda/environments/cuml_dev_cuda11.4.yml new file mode 100644 index 0000000000..65fe223ccb --- /dev/null +++ b/conda/environments/cuml_dev_cuda11.4.yml @@ -0,0 +1,42 @@ +name: cuml_dev +channels: +- rapidsai +- nvidia +- rapidsai-nightly +- conda-forge +dependencies: +- cudatoolkit=11.4 +- rapids-build-env=21.10.* +- rapids-notebook-env=21.10.* +- rapids-doc-env=21.10.* +- cudf=21.10.* +- rmm=21.10.* +- libcumlprims=21.10.* +- dask-cudf=21.10.* +- dask-cuda=21.10.* +- ucx-py=0.21 +- ucx-proc=*=gpu +- dask-ml +- doxygen>=1.8.20 +- libfaiss>=1.7.0 +- faiss-proc=*=cuda +- umap-learn +- scikit-learn=0.23.1 +- treelite=2.0.0 +- statsmodels +- seaborn +- hdbscan +- nltk +- pip +- pip: + - sphinx_markdown_tables + - git+https://github.com/dask/dask.git@main + - git+https://github.com/dask/distributed.git@main + +# rapids-build-env, notebook-env and doc-env are defined in +# https://docs.rapids.ai/maintainers/depmgmt/ + +# To install different versions of packages contained in those meta packages, +# it is recommended to remove those meta packages (without removing the actual +# packages contained in the environment) first with: +# conda remove --force rapids-build-env rapids-notebook-env rapids-doc-env