Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[doc] fix the cudf installation [skip ci] (#9106)

Browse Source
This commit is contained in:
Bobby Wang 2023-04-28 19:43:58 +08:00 committed by GitHub
parent 17ff471616
commit e922004329
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
doc/tutorials/spark_estimator.rst
View File

@ -108,8 +108,8 @@ virtualenv and pip:
python -m venv xgboost_env python -m venv xgboost_env
source xgboost_env/bin/activate source xgboost_env/bin/activate
pip install pyarrow pandas venv-pack xgboost pip install pyarrow pandas venv-pack xgboost
# https://rapids.ai/pip.html#install # https://docs.rapids.ai/install#pip-install
pip install cudf-cu11 --extra-index-url=https://pypi.ngc.nvidia.com pip install cudf-cu11 --extra-index-url=https://pypi.nvidia.com
venv-pack -o xgboost_env.tar.gz venv-pack -o xgboost_env.tar.gz
With Conda: With Conda:
@ -241,7 +241,7 @@ additional spark configurations and dependencies:
--master spark://<master-ip>:7077 \ --master spark://<master-ip>:7077 \
--conf spark.executor.resource.gpu.amount=1 \ --conf spark.executor.resource.gpu.amount=1 \
--conf spark.task.resource.gpu.amount=1 \ --conf spark.task.resource.gpu.amount=1 \
--packages com.nvidia:rapids-4-spark_2.12:22.08.0 \ --packages com.nvidia:rapids-4-spark_2.12:23.04.0 \
--conf spark.plugins=com.nvidia.spark.SQLPlugin \ --conf spark.plugins=com.nvidia.spark.SQLPlugin \
--conf spark.sql.execution.arrow.maxRecordsPerBatch=1000000 \ --conf spark.sql.execution.arrow.maxRecordsPerBatch=1000000 \
--archives xgboost_env.tar.gz#environment \ --archives xgboost_env.tar.gz#environment \