diff --git a/examples/XGBoost-Examples/agaricus/notebooks/python/agaricus-gpu.ipynb b/examples/XGBoost-Examples/agaricus/notebooks/python/agaricus-gpu.ipynb
index d6d14583d..76852e2f9 100644
--- a/examples/XGBoost-Examples/agaricus/notebooks/python/agaricus-gpu.ipynb
+++ b/examples/XGBoost-Examples/agaricus/notebooks/python/agaricus-gpu.ipynb
@@ -197,7 +197,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "params['features_col'] = features\n",
     "params['label_col'] = label\n",
diff --git a/examples/XGBoost-Examples/mortgage/notebooks/python/MortgageETL+XGBoost.ipynb b/examples/XGBoost-Examples/mortgage/notebooks/python/MortgageETL+XGBoost.ipynb
index 48830285d..19b0c211d 100644
--- a/examples/XGBoost-Examples/mortgage/notebooks/python/MortgageETL+XGBoost.ipynb
+++ b/examples/XGBoost-Examples/mortgage/notebooks/python/MortgageETL+XGBoost.ipynb
@@ -933,7 +933,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "params['features_col'] = features\n",
     "params['label_col'] = label\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
index 54c975af4..97de20728 100644
--- a/examples/XGBoost-Examples/mortgage/notebooks/python/cv-mortgage-gpu.ipynb
+++ b/examples/XGBoost-Examples/mortgage/notebooks/python/cv-mortgage-gpu.ipynb
@@ -191,7 +191,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "\n",
     "params['features_col'] = features\n",
diff --git a/examples/XGBoost-Examples/mortgage/notebooks/python/mortgage-gpu.ipynb b/examples/XGBoost-Examples/mortgage/notebooks/python/mortgage-gpu.ipynb
index 9312fc183..e813b8072 100644
--- a/examples/XGBoost-Examples/mortgage/notebooks/python/mortgage-gpu.ipynb
+++ b/examples/XGBoost-Examples/mortgage/notebooks/python/mortgage-gpu.ipynb
@@ -235,7 +235,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "params['features_col'] = features\n",
     "params['label_col'] = label\n",
diff --git a/examples/XGBoost-Examples/taxi/notebooks/python/cv-taxi-gpu.ipynb b/examples/XGBoost-Examples/taxi/notebooks/python/cv-taxi-gpu.ipynb
index 19bedab47..24fe12492 100644
--- a/examples/XGBoost-Examples/taxi/notebooks/python/cv-taxi-gpu.ipynb
+++ b/examples/XGBoost-Examples/taxi/notebooks/python/cv-taxi-gpu.ipynb
@@ -182,7 +182,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "params['features_col'] = features\n",
     "params['label_col'] = label\n",
diff --git a/examples/XGBoost-Examples/taxi/notebooks/python/taxi-gpu.ipynb b/examples/XGBoost-Examples/taxi/notebooks/python/taxi-gpu.ipynb
index ce3ee91a1..a17282842 100644
--- a/examples/XGBoost-Examples/taxi/notebooks/python/taxi-gpu.ipynb
+++ b/examples/XGBoost-Examples/taxi/notebooks/python/taxi-gpu.ipynb
@@ -213,7 +213,7 @@
     "    \"tree_method\": \"gpu_hist\",\n",
     "    \"grow_policy\": \"depthwise\",\n",
     "    \"num_workers\": 1,\n",
-    "    \"use_gpu\": \"true\",\n",
+    "    \"device\": \"cuda\",\n",
     "}\n",
     "params['features_col'] = features\n",
     "params['label_col'] = label\n",