update the main branch for 2506 release (#550)

nvliyuan · web-flow · commit 626416c74570 · 2025-06-30T13:52:58.000+08:00
update the main branch for 2506 release.
Please create a merge commit, not squash.
diff --git a/docs/get-started/xgboost-examples/csp/databricks/databricks.md b/docs/get-started/xgboost-examples/csp/databricks/databricks.md
@@ -21,7 +21,7 @@ Navigate to your home directory in the UI and select **Create** > **File** from
 create an `init.sh` scripts with contents:   
    ```bash
    #!/bin/bash
-   sudo wget -O /databricks/jars/rapids-4-spark_2.12-25.04.0.jar https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.04.0/rapids-4-spark_2.12-25.04.0.jar
+   sudo wget -O /databricks/jars/rapids-4-spark_2.12-25.06.0.jar https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.06.0/rapids-4-spark_2.12-25.06.0.jar
    ```
 1. Select the Databricks Runtime Version from one of the supported runtimes specified in the
    Prerequisites section.
@@ -68,7 +68,7 @@ create an `init.sh` scripts with contents:
     ```bash
     spark.rapids.sql.python.gpu.enabled true
     spark.python.daemon.module rapids.daemon_databricks
-    spark.executorEnv.PYTHONPATH /databricks/jars/rapids-4-spark_2.12-25.04.0.jar:/databricks/spark/python
+    spark.executorEnv.PYTHONPATH /databricks/jars/rapids-4-spark_2.12-25.06.0.jar:/databricks/spark/python
     ```
    Note that since python memory pool require installing the cudf library, so you need to install cudf library in 
    each worker nodes `pip install cudf-cu11 --extra-index-url=https://pypi.nvidia.com` or disable python memory pool
diff --git a/docs/get-started/xgboost-examples/csp/databricks/init.sh b/docs/get-started/xgboost-examples/csp/databricks/init.sh
@@ -1,4 +1,5 @@
-# Copyright (c) 2024, NVIDIA CORPORATION.
+#
+# Copyright (c) 2025, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -11,11 +12,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+#
 
 sudo rm -f /databricks/jars/spark--maven-trees--ml--10.x--xgboost-gpu--ml.dmlc--xgboost4j-gpu_2.12--ml.dmlc__xgboost4j-gpu_2.12__1.5.2.jar
 sudo rm -f /databricks/jars/spark--maven-trees--ml--10.x--xgboost-gpu--ml.dmlc--xgboost4j-spark-gpu_2.12--ml.dmlc__xgboost4j-spark-gpu_2.12__1.5.2.jar
 
-sudo wget -O /databricks/jars/rapids-4-spark_2.12-25.04.0.jar https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.04.0/rapids-4-spark_2.12-25.04.0.jar
+sudo wget -O /databricks/jars/rapids-4-spark_2.12-25.06.0.jar https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.06.0/rapids-4-spark_2.12-25.06.0.jar
 sudo wget -O /databricks/jars/xgboost4j-gpu_2.12-1.7.1.jar https://repo1.maven.org/maven2/ml/dmlc/xgboost4j-gpu_2.12/1.7.1/xgboost4j-gpu_2.12-1.7.1.jar
 sudo wget -O /databricks/jars/xgboost4j-spark-gpu_2.12-1.7.1.jar https://repo1.maven.org/maven2/ml/dmlc/xgboost4j-spark-gpu_2.12/1.7.1/xgboost4j-spark-gpu_2.12-1.7.1.jar
 ls -ltr
diff --git a/docs/get-started/xgboost-examples/csp/dataproc/gcp.md b/docs/get-started/xgboost-examples/csp/dataproc/gcp.md
@@ -187,4 +187,5 @@ gcloud dataproc clusters create $CLUSTER_NAME  \
     --subnet=default 
 ```
 
-The new cluster should be up and running within 3-4 minutes!
+The new cluster should be up and running within 3-4 minutes!
+
diff --git a/docs/get-started/xgboost-examples/on-prem-cluster/kubernetes-scala.md b/docs/get-started/xgboost-examples/on-prem-cluster/kubernetes-scala.md
@@ -40,7 +40,7 @@ export SPARK_DOCKER_IMAGE=<gpu spark docker image repo and name>
 export SPARK_DOCKER_TAG=<spark docker image tag>
 
 pushd ${SPARK_HOME}
-wget https://github.com/NVIDIA/spark-rapids-examples/raw/branch-25.04/dockerfile/Dockerfile
+wget https://github.com/NVIDIA/spark-rapids-examples/raw/branch-25.06/dockerfile/Dockerfile
 
 # Optionally install additional jars into ${SPARK_HOME}/jars/
 
diff --git a/docs/get-started/xgboost-examples/prepare-package-data/preparation-python.md b/docs/get-started/xgboost-examples/prepare-package-data/preparation-python.md
@@ -5,7 +5,7 @@ For simplicity export the location to these jars. All examples assume the packag
 ### Download the jars
 
 Download the RAPIDS Accelerator for Apache Spark plugin jar
-  * [RAPIDS Spark Package](https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.04.0/rapids-4-spark_2.12-25.04.0.jar)
+  * [RAPIDS Spark Package](https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.06.0/rapids-4-spark_2.12-25.06.0.jar)
 
 ### Build XGBoost Python Examples
 
diff --git a/docs/get-started/xgboost-examples/prepare-package-data/preparation-scala.md b/docs/get-started/xgboost-examples/prepare-package-data/preparation-scala.md
@@ -5,7 +5,7 @@ For simplicity export the location to these jars. All examples assume the packag
 ### Download the jars
 
 1. Download the RAPIDS Accelerator for Apache Spark plugin jar
-   * [RAPIDS Spark Package](https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.04.0/rapids-4-spark_2.12-25.04.0.jar)
+   * [RAPIDS Spark Package](https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.06.0/rapids-4-spark_2.12-25.06.0.jar)
 
 ### Build XGBoost Scala Examples
 
diff --git a/docs/img/guides/tpcds.png b/docs/img/guides/tpcds.png
diff --git a/examples/ML+DL-Examples/Optuna-Spark/README.md b/examples/ML+DL-Examples/Optuna-Spark/README.md
@@ -147,8 +147,8 @@ We use [RAPIDS](https://docs.rapids.ai/install/#get-rapids) for GPU-accelerated
 ``` shell
 sudo apt install libmysqlclient-dev
 
-conda create -n rapids-25.04 -c rapidsai -c conda-forge -c nvidia  \
-    cudf=25.04 cuml=25.04 python=3.10 'cuda-version>=12.0,<=12.5'
+conda create -n rapids-25.06 -c rapidsai -c conda-forge -c nvidia  \
+    cudf=25.06 cuml=25.06 python=3.10 'cuda-version>=12.0,<=12.5'
 conda activate optuna-spark
 pip install mysqlclient
 pip install optuna joblib joblibspark ipywidgets
diff --git a/examples/ML+DL-Examples/Optuna-Spark/optuna-examples/databricks/init_optuna.sh b/examples/ML+DL-Examples/Optuna-Spark/optuna-examples/databricks/init_optuna.sh
@@ -1,5 +1,19 @@
 #!/bin/bash
-# Copyright (c) 2024, NVIDIA CORPORATION.
+#
+# Copyright (c) 2025, NVIDIA CORPORATION.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
 
 set -x
 
@@ -41,7 +55,7 @@ fi
 
 
 # rapids import
-SPARK_RAPIDS_VERSION=25.04.0
+SPARK_RAPIDS_VERSION=25.06.0
 curl -L https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/${SPARK_RAPIDS_VERSION}/rapids-4-spark_2.12-${SPARK_RAPIDS_VERSION}.jar -o \
     /databricks/jars/rapids-4-spark_2.12-${SPARK_RAPIDS_VERSION}.jar
 
diff --git a/examples/ML+DL-Examples/Optuna-Spark/optuna-examples/databricks/start_cluster.sh b/examples/ML+DL-Examples/Optuna-Spark/optuna-examples/databricks/start_cluster.sh
@@ -1,5 +1,19 @@
 #!/bin/bash
-# Copyright (c) 2024, NVIDIA CORPORATION.
+#
+# Copyright (c) 2025, NVIDIA CORPORATION.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
 
 if [[ -z ${INIT_PATH} ]]; then
     echo "Please export INIT_PATH per README.md"
@@ -12,7 +26,7 @@ json_config=$(cat <<EOF
     "spark_version": "13.3.x-gpu-ml-scala2.12",
     "spark_conf": {
         "spark.task.resource.gpu.amount": "1",
-        "spark.executorEnv.PYTHONPATH": "/databricks/jars/rapids-4-spark_2.12-25.04.0.jar:/databricks/spark/python:/databricks/python3",
+        "spark.executorEnv.PYTHONPATH": "/databricks/jars/rapids-4-spark_2.12-25.06.0.jar:/databricks/spark/python:/databricks/python3",
         "spark.executor.cores": "8",
         "spark.rapids.memory.gpu.minAllocFraction": "0.0001",
         "spark.plugins": "com.nvidia.spark.SQLPlugin",
diff --git a/examples/ML+DL-Examples/Optuna-Spark/optuna-examples/optuna-dataframe.ipynb b/examples/ML+DL-Examples/Optuna-Spark/optuna-examples/optuna-dataframe.ipynb
@@ -451,7 +451,7 @@
    ],
    "source": [
     "def get_rapids_jar():\n",
-    "    SPARK_RAPIDS_VERSION = \"25.04.0\"\n",
+    "    SPARK_RAPIDS_VERSION = \"25.06.0\"\n",
     "    rapids_jar = f\"rapids-4-spark_2.12-{SPARK_RAPIDS_VERSION}.jar\"\n",
     "    if not os.path.exists(rapids_jar):\n",
     "        print(\"Downloading Spark Rapids jar\")\n",
diff --git a/examples/ML+DL-Examples/Spark-Rapids-ML/pca/README.md b/examples/ML+DL-Examples/Spark-Rapids-ML/pca/README.md
@@ -10,7 +10,7 @@ Please refer to the Spark-Rapids-ML [README](https://github.com/NVIDIA/spark-rap
 ## Download RAPIDS Jar from Maven Central
 
 Download the [Spark-Rapids plugin](https://nvidia.github.io/spark-rapids/docs/download.html#download-rapids-accelerator-for-apache-spark-v24081).  
-For Spark-RAPIDS-ML version 25.04.0, download the RAPIDS jar from Maven Central: [rapids-4-spark_2.12-25.04.0.jar](https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.04.0/rapids-4-spark_2.12-25.04.0.jar). 
+For Spark-RAPIDS-ML version 25.06.0, download the RAPIDS jar from Maven Central: [rapids-4-spark_2.12-25.06.0.jar](https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.06.0/rapids-4-spark_2.12-25.06.0.jar). 
 
 ## Running the Notebooks
 
diff --git a/examples/ML+DL-Examples/Spark-Rapids-ML/pca/notebooks/pca.ipynb b/examples/ML+DL-Examples/Spark-Rapids-ML/pca/notebooks/pca.ipynb
@@ -57,7 +57,7 @@
     "    import os\n",
     "    import requests\n",
     "\n",
-    "    SPARK_RAPIDS_VERSION = \"25.04.0\"\n",
+    "    SPARK_RAPIDS_VERSION = \"25.06.0\"\n",
     "    rapids_jar = f\"rapids-4-spark_2.12-{SPARK_RAPIDS_VERSION}.jar\"\n",
     "    if not os.path.exists(rapids_jar):\n",
     "        print(\"Downloading spark rapids jar\")\n",
diff --git a/examples/SQL+DF-Examples/micro-benchmarks/notebooks/micro-benchmarks-gpu.ipynb b/examples/SQL+DF-Examples/micro-benchmarks/notebooks/micro-benchmarks-gpu.ipynb
@@ -22,7 +22,7 @@
     "import os\n",
     "# Change to your cluster ip:port and directories\n",
     "SPARK_MASTER_URL = os.getenv(\"SPARK_MASTER_URL\", \"spark:your-ip:port\")\n",
-    "RAPIDS_JAR = os.getenv(\"RAPIDS_JAR\", \"/your-path/rapids-4-spark_2.12-25.04.0.jar\")\n"
+    "RAPIDS_JAR = os.getenv(\"RAPIDS_JAR\", \"/your-path/rapids-4-spark_2.12-25.06.0.jar\")\n"
    ]
   },
   {
diff --git a/examples/SQL+DF-Examples/tpcds/README.md b/examples/SQL+DF-Examples/tpcds/README.md
@@ -18,11 +18,11 @@ This notebook can be opened and executed using standard
 It can also be opened and evaluated on hosted Notebook environments. Use the link below to launch on
 Google Colab and connect it to a [GPU instance](https://research.google.com/colaboratory/faq.html).
 
- <a target="_blank" href="https://colab.research.google.com/github/NVIDIA/spark-rapids-examples/blob/branch-25.04/examples/SQL%2BDF-Examples/tpcds/notebooks/TPCDS-SF10.ipynb">
+ <a target="_blank" href="https://colab.research.google.com/github/NVIDIA/spark-rapids-examples/blob/branch-25.06/examples/SQL%2BDF-Examples/tpcds/notebooks/TPCDS-SF10.ipynb">
   <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
 </a>
 
 Here is the bar chart from a recent execution on Google Colab's T4 High RAM instance using
-RAPIDS Spark 25.04.0 with Apache Spark 3.5.0
+RAPIDS Spark 25.06.0 with Apache Spark 3.5.0
 
 ![tpcds-speedup](/docs/img/guides/tpcds.png)
diff --git a/examples/SQL+DF-Examples/tpcds/notebooks/TPCDS-SF10.ipynb b/examples/SQL+DF-Examples/tpcds/notebooks/TPCDS-SF10.ipynb
@@ -29,8 +29,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "spark_version='3.5.0'\n",
-    "rapids_version='25.04.0'"
+    "spark_version='3.5.5'\n",
+    "rapids_version='25.06.0'"
    ]
   },
   {
@@ -52,10 +52,10 @@
    "outputs": [],
    "source": [
     "%pip install --quiet \\\n",
-    "  tpcds_pyspark \\\n",
+    "  tpcds_pyspark==1.0.5 \\\n",
     "  pyspark=={spark_version} \\\n",
-    "  sparkmeasure \\\n",
     "  pandas \\\n",
+    "  sparkmeasure==0.23.2 \\\n",
     "  matplotlib"
    ]
   },
@@ -468,7 +468,9 @@
    },
    "outputs": [],
    "source": [
-    "res.plot(title='TPC-DS query elapsedTime on CPU vs GPU (lower is better)', kind='bar', x='query', y=['elapsedTime_cpu', 'elapsedTime_gpu'])"
+    "res.plot(title='TPC-DS query elapsedTime on CPU vs GPU (lower is better)', \n",
+    "         kind='bar', x='query', y=['elapsedTime_cpu', 'elapsedTime_gpu'],\n",
+    "         color=['blue', '#76B900'])"
    ]
   },
   {
@@ -494,7 +496,8 @@
    },
    "outputs": [],
    "source": [
-    "res.plot(title='Speedup factors of TPC-DS queries on GPU', kind='bar', x='query', y='speedup' )"
+    "res.plot(title='Speedup factors of TPC-DS queries on GPU', kind='bar', \n",
+    "         x='query', y='speedup', color='#76B900')"
    ]
   },
   {
@@ -597,7 +600,7 @@
    "provenance": []
   },
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": "notebooks-5X_G7DxN",
    "language": "python",
    "name": "python3"
   },
diff --git a/examples/UDF-Examples/RAPIDS-accelerated-UDFs/Dockerfile b/examples/UDF-Examples/RAPIDS-accelerated-UDFs/Dockerfile
@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2021-2024, NVIDIA CORPORATION. All rights reserved.
+# Copyright (c) 2021-2025, NVIDIA CORPORATION. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -41,7 +41,7 @@ RUN dnf --enablerepo=powertools install -y \
   && alternatives --set python /usr/bin/python3
 
 # 3.22.3: CUDA architecture 'native' support + flexible CMAKE_<LANG>_*_LAUNCHER for ccache
-ARG CMAKE_VERSION=3.28.6
+ARG CMAKE_VERSION=3.30.4
 # default x86_64 from x86 build, aarch64 cmake for arm build
 ARG CMAKE_ARCH=x86_64
 RUN cd /usr/local && wget --quiet https://github.com/Kitware/CMake/releases/download/v${CMAKE_VERSION}/cmake-${CMAKE_VERSION}-linux-${CMAKE_ARCH}.tar.gz && \
diff --git a/examples/UDF-Examples/RAPIDS-accelerated-UDFs/README.md b/examples/UDF-Examples/RAPIDS-accelerated-UDFs/README.md
@@ -186,7 +186,7 @@ then do the following inside the Docker container.
 
 ### Get jars from Maven Central
 
-[rapids-4-spark_2.12-25.04.0.jar](https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.04.0/rapids-4-spark_2.12-25.04.0.jar)
+[rapids-4-spark_2.12-25.06.0.jar](https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.06.0/rapids-4-spark_2.12-25.06.0.jar)
 
 
 ### Launch a local mode Spark
diff --git a/examples/UDF-Examples/RAPIDS-accelerated-UDFs/pom.xml b/examples/UDF-Examples/RAPIDS-accelerated-UDFs/pom.xml
@@ -25,7 +25,7 @@
         user defined functions for use with the RAPIDS Accelerator
         for Apache Spark
     </description>
-    <version>25.04.0-SNAPSHOT</version>
+    <version>25.06.0-SNAPSHOT</version>
 
     <properties>
         <maven.compiler.source>1.8</maven.compiler.source>
@@ -37,7 +37,7 @@
         <cuda.version>cuda11</cuda.version>
         <scala.binary.version>2.12</scala.binary.version>
         <!-- Depends on release version, Snapshot version is not published to the Maven Central -->
-        <rapids4spark.version>25.04.0</rapids4spark.version>
+        <rapids4spark.version>25.06.0</rapids4spark.version>
         <spark.version>3.1.1</spark.version>
         <scala.version>2.12.15</scala.version>
         <udf.native.build.path>${project.build.directory}/cpp-build</udf.native.build.path>
diff --git a/examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/cpp/CMakeLists.txt b/examples/UDF-Examples/RAPIDS-accelerated-UDFs/src/main/cpp/CMakeLists.txt
@@ -14,9 +14,14 @@
 # limitations under the License.
 #=============================================================================
 
-cmake_minimum_required(VERSION 3.28.6 FATAL_ERROR)
+# Keep the same with https://github.com/rapidsai/rapids-cmake/blob/branch-xx.yy/RAPIDS.cmake
+# the branch-xx.xx in the above line is the branch version
+cmake_minimum_required(VERSION 3.30.4 FATAL_ERROR)
 
-file(DOWNLOAD https://raw.githubusercontent.com/rapidsai/rapids-cmake/branch-25.04/RAPIDS.cmake
+# set to the rapids-cmake version
+set(rapids-cmake-version "25.06")
+
+file(DOWNLOAD https://raw.githubusercontent.com/rapidsai/rapids-cmake/branch-${rapids-cmake-version}/RAPIDS.cmake
      ${CMAKE_BINARY_DIR}/RAPIDS.cmake)
 include(${CMAKE_BINARY_DIR}/RAPIDS.cmake)
 
@@ -32,7 +37,7 @@ if(DEFINED GPU_ARCHS)
 endif()
 rapids_cuda_init_architectures(UDFEXAMPLESJNI)
 
-project(UDFEXAMPLESJNI VERSION 25.04.0 LANGUAGES C CXX CUDA)
+project(UDFEXAMPLESJNI VERSION 25.06.0 LANGUAGES C CXX CUDA)
 
 option(PER_THREAD_DEFAULT_STREAM "Build with per-thread default stream" OFF)
 option(BUILD_UDF_BENCHMARKS "Build the benchmarks" OFF)
@@ -84,10 +89,10 @@ set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -w --expt-extended-lambda --expt-relax
 set(CUDA_USE_STATIC_CUDA_RUNTIME ON)
 
 rapids_cpm_init()
-rapids_cpm_find(cudf 25.04.00
+rapids_cpm_find(cudf 25.06.00
         CPM_ARGS
         GIT_REPOSITORY  https://github.com/rapidsai/cudf.git
-        GIT_TAG         branch-25.04
+        GIT_TAG         branch-25.06
         GIT_SHALLOW     TRUE
         SOURCE_SUBDIR   cpp
         OPTIONS         "BUILD_TESTS OFF"
@@ -173,4 +178,4 @@ target_compile_definitions(udfexamplesjni
 ###################################################################################################
 # - link libraries --------------------------------------------------------------------------------
 
-target_link_libraries(udfexamplesjni cudf::cudf)
+target_link_libraries(udfexamplesjni cudf::cudf)
diff --git a/examples/XGBoost-Examples/agaricus/notebooks/python/agaricus-gpu.ipynb b/examples/XGBoost-Examples/agaricus/notebooks/python/agaricus-gpu.ipynb
@@ -108,8 +108,8 @@
     "conf.set(\"spark.executor.resource.gpu.amount\", \"1\")\n",
     "conf.set(\"spark.rapids.sql.concurrentGpuTasks\", concurrentGpuTasks)\n",
     "conf.set(\"spark.rapids.memory.pinnedPool.size\", pinnedPoolSize)\n",
-    "# since pyspark and xgboost share the same GPU, we need to allocate some memory to xgboost to avoid GPU OOM while training \n",
-    "conf.set(\"spark.rapids.memory.gpu.allocFraction\",\"0.7\")\n",
+    "# since pyspark and xgboost share the same GPU, we disable RMM to avoid GPU OOM while training \n",
+    "conf.set(\"spark.rapids.memory.gpu.pool\", \"NONE\")\n",
     "conf.set(\"spark.locality.wait\",\"0\")\n",
     "##############note: only support value=1 https://github.com/dmlc/xgboost/blame/master/python-package/xgboost/spark/core.py#L370-L374\n",
     "conf.set(\"spark.task.resource.gpu.amount\", 1) \n",
diff --git a/examples/XGBoost-Examples/mortgage/notebooks/python/MortgageETL+XGBoost.ipynb b/examples/XGBoost-Examples/mortgage/notebooks/python/MortgageETL+XGBoost.ipynb
@@ -90,8 +90,8 @@
     "conf.set(\"spark.rapids.memory.pinnedPool.size\", pinnedPoolSize)\n",
     "##############note: only support value=1 see https://github.com/dmlc/xgboost/blame/master/python-package/xgboost/spark/core.py#L370-L374\n",
     "conf.set(\"spark.task.resource.gpu.amount\", 1) \n",
-    "# since pyspark and xgboost share the same GPU, we need to allocate some memory to xgboost to avoid GPU OOM while training \n",
-    "conf.set(\"spark.rapids.memory.gpu.allocFraction\",\"0.6\")\n",
+    "# since pyspark and xgboost share the same GPU, we disable RMM to avoid GPU OOM while training \n",
+    "conf.set(\"spark.rapids.memory.gpu.pool\", \"NONE\")\n",
     "conf.set(\"spark.rapids.sql.enabled\", \"true\") \n",
     "conf.set(\"spark.plugins\", \"com.nvidia.spark.SQLPlugin\")\n",
     "conf.set(\"spark.sql.cache.serializer\",\"com.nvidia.spark.ParquetCachedBatchSerializer\")\n",
diff --git a/examples/XGBoost-Examples/mortgage/notebooks/python/MortgageETL.ipynb b/examples/XGBoost-Examples/mortgage/notebooks/python/MortgageETL.ipynb
@@ -9,15 +9,15 @@
     "Dataset is derived from Fannie Mae’s [Single-Family Loan Performance Data](http://www.fanniemae.com/portal/funding-the-market/data/loan-performance-data.html) with all rights reserved by Fannie Mae. Refer to these [instructions](https://github.com/NVIDIA/spark-rapids-examples/blob/branch-24.12/docs/get-started/xgboost-examples/dataset/mortgage.md) to download the dataset.\n",
     "\n",
     "### 2. Download needed jars\n",
-    "* [rapids-4-spark_2.12-25.04.0.jar](https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.04.0/rapids-4-spark_2.12-25.04.0.jar)\n",
+    "* [rapids-4-spark_2.12-25.06.0.jar](https://repo1.maven.org/maven2/com/nvidia/rapids-4-spark_2.12/25.06.0/rapids-4-spark_2.12-25.06.0.jar)\n",
     "\n",
     "\n",
     "### 3. Start Spark Standalone\n",
     "Before running the script, please setup Spark standalone mode\n",
     "\n",
     "### 4. Add ENV\n",
     "```\n",
-    "$ export SPARK_JARS=rapids-4-spark_2.12-25.04.0.jar\n",
+    "$ export SPARK_JARS=rapids-4-spark_2.12-25.06.0.jar\n",
     "$ export PYSPARK_DRIVER_PYTHON=jupyter \n",
     "$ export PYSPARK_DRIVER_PYTHON_OPTS=notebook\n",
     "```\n",
diff --git a/examples/XGBoost-Examples/mortgage/notebooks/python/cv-mortgage-gpu.ipynb b/examples/XGBoost-Examples/mortgage/notebooks/python/cv-mortgage-gpu.ipynb
@@ -95,7 +95,8 @@
     "conf.set(\"spark.executor.resource.gpu.amount\", \"1\")\n",
     "conf.set(\"spark.rapids.sql.concurrentGpuTasks\", concurrentGpuTasks)\n",
     "conf.set(\"spark.rapids.memory.pinnedPool.size\", pinnedPoolSize)\n",
-    "conf.set(\"spark.rapids.memory.gpu.allocFraction\",\"0.7\")\n",
+    "# since pyspark and xgboost share the same GPU, we disable RMM to avoid GPU OOM while training \n",
+    "conf.set(\"spark.rapids.memory.gpu.pool\", \"NONE\")\n",
     "conf.set(\"spark.locality.wait\",\"0\")\n",
     "##############note: only support value=1 https://github.com/dmlc/xgboost/blame/master/python-package/xgboost/spark/core.py#L370-L374\n",
     "conf.set(\"spark.task.resource.gpu.amount\", 1) \n",
diff --git a/examples/XGBoost-Examples/mortgage/notebooks/python/mortgage-gpu.ipynb b/examples/XGBoost-Examples/mortgage/notebooks/python/mortgage-gpu.ipynb
@@ -118,8 +118,8 @@
     "conf.set(\"spark.rapids.memory.pinnedPool.size\", pinnedPoolSize)\n",
     "##############note: only support value=1 see https://github.com/dmlc/xgboost/blame/master/python-package/xgboost/spark/core.py#L370-L374\n",
     "conf.set(\"spark.task.resource.gpu.amount\", 1) \n",
-    "# since pyspark and xgboost share the same GPU, we need to allocate some memory to xgboost to avoid GPU OOM while training \n",
-    "conf.set(\"spark.rapids.memory.gpu.allocFraction\",\"0.6\")\n",
+    "# since pyspark and xgboost share the same GPU, we disable RMM to avoid GPU OOM while training \n",
+    "conf.set(\"spark.rapids.memory.gpu.pool\", \"NONE\")\n",
     "conf.set(\"spark.rapids.sql.enabled\", \"true\") \n",
     "conf.set(\"spark.plugins\", \"com.nvidia.spark.SQLPlugin\")\n",
     "conf.set(\"spark.sql.cache.serializer\",\"com.nvidia.spark.ParquetCachedBatchSerializer\")\n",
diff --git a/examples/XGBoost-Examples/mortgage/notebooks/scala/mortgage-ETL.ipynb b/examples/XGBoost-Examples/mortgage/notebooks/scala/mortgage-ETL.ipynb
diff --git a/examples/XGBoost-Examples/taxi/notebooks/python/cv-taxi-gpu.ipynb b/examples/XGBoost-Examples/taxi/notebooks/python/cv-taxi-gpu.ipynb
diff --git a/examples/XGBoost-Examples/taxi/notebooks/python/taxi-ETL.ipynb b/examples/XGBoost-Examples/taxi/notebooks/python/taxi-ETL.ipynb
diff --git a/examples/XGBoost-Examples/taxi/notebooks/python/taxi-gpu.ipynb b/examples/XGBoost-Examples/taxi/notebooks/python/taxi-gpu.ipynb
diff --git a/examples/XGBoost-Examples/taxi/notebooks/scala/taxi-ETL.ipynb b/examples/XGBoost-Examples/taxi/notebooks/scala/taxi-ETL.ipynb
diff --git a/tools/databricks/README.md b/tools/databricks/README.md
diff --git a/tools/databricks/[RAPIDS Accelerator for Apache Spark] Profiling Tool Notebook Template.ipynb b/tools/databricks/[RAPIDS Accelerator for Apache Spark] Profiling Tool Notebook Template.ipynb
diff --git a/tools/databricks/[RAPIDS Accelerator for Apache Spark] Qualification Tool Notebook Template.ipynb b/tools/databricks/[RAPIDS Accelerator for Apache Spark] Qualification Tool Notebook Template.ipynb
