From 03f3879b713a00bf800f77757276afa3f1e95d39 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Sat, 10 Jun 2023 04:09:27 +0800 Subject: [PATCH] [backport] [doc] fix the cudf installation [skip ci] (#9106) (#9286) Co-authored-by: Bobby Wang --- doc/tutorials/spark_estimator.rst | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/doc/tutorials/spark_estimator.rst b/doc/tutorials/spark_estimator.rst index e0acc7454..c8cad2919 100644 --- a/doc/tutorials/spark_estimator.rst +++ b/doc/tutorials/spark_estimator.rst @@ -107,8 +107,8 @@ virtualenv and pip: python -m venv xgboost_env source xgboost_env/bin/activate pip install pyarrow pandas venv-pack xgboost - # https://rapids.ai/pip.html#install - pip install cudf-cu11 --extra-index-url=https://pypi.ngc.nvidia.com + # https://docs.rapids.ai/install#pip-install + pip install cudf-cu11 --extra-index-url=https://pypi.nvidia.com venv-pack -o xgboost_env.tar.gz With Conda: @@ -240,7 +240,7 @@ additional spark configurations and dependencies: --master spark://:7077 \ --conf spark.executor.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.sql.execution.arrow.maxRecordsPerBatch=1000000 \ --archives xgboost_env.tar.gz#environment \