mirror of
https://github.com/Azure/MachineLearningNotebooks.git
synced 2025-12-25 10:00:15 -05:00
36 lines
616 B
YAML
36 lines
616 B
YAML
name: rapids
|
|
channels:
|
|
- nvidia
|
|
- numba
|
|
- conda-forge
|
|
- rapidsai
|
|
- defaults
|
|
- pytorch
|
|
|
|
dependencies:
|
|
- arrow-cpp=0.12.0
|
|
- bokeh
|
|
- cffi=1.11.5
|
|
- cmake=3.12
|
|
- cuda92
|
|
- cython==0.29
|
|
- dask=1.1.1
|
|
- distributed=1.25.3
|
|
- faiss-gpu=1.5.0
|
|
- numba=0.42
|
|
- numpy=1.15.4
|
|
- nvstrings
|
|
- pandas=0.23.4
|
|
- pyarrow=0.12.0
|
|
- scikit-learn
|
|
- scipy
|
|
- cudf
|
|
- cuml
|
|
- python=3.6.2
|
|
- jupyterlab
|
|
- pip:
|
|
- file:/rapids/xgboost/python-package/dist/xgboost-0.81-py3-none-any.whl
|
|
- git+https://github.com/rapidsai/dask-xgboost@dask-cudf
|
|
- git+https://github.com/rapidsai/dask-cudf@master
|
|
- git+https://github.com/rapidsai/dask-cuda@master
|