From 871dcbc7e1cf891dc26139176c235fb1bb2dbee8 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Thu, 13 Jul 2023 18:00:06 +0800 Subject: [PATCH] Reviewer's comment. --- doc/parameter.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/doc/parameter.rst b/doc/parameter.rst index 56a30779936e..d628d161b515 100644 --- a/doc/parameter.rst +++ b/doc/parameter.rst @@ -68,8 +68,8 @@ General Parameters + ``cpu``: Use CPU. + ``cuda``: Use a GPU (CUDA device). + ``cuda:``: ```` is an integer that specifies the ordinal of the GPU (which GPU do you want to use if you have more than one devices). - + ``gpu``: Same as ``cuda``. - + ``gpu:``: Same as ``cuda:``. + + ``gpu``: Default GPU device selection from the list of available and supported devices. Only ``cuda`` devices are supported currently. + + ``gpu:``: Default GPU device selection from the list of available and supported devices. Only ``cuda`` devices are supported currently. Parameters for Tree Booster ===========================