[BLOCKING] [jvm-packages] add gpu_hist and enable gpu scheduling (#5171)
* [jvm-packages] add gpu_hist tree method * change updater hist to grow_quantile_histmaker * add gpu scheduling * pass correct parameters to xgboost library * remove debug info * add use.cuda for pom * add CI for gpu_hist for jvm * add gpu unit tests * use gpu node to build jvm * use nvidia-docker * Add CLI interface to create_jni.py using argparse Co-authored-by: Hyunsu Cho <chohyu01@cs.washington.edu>
This commit is contained in:
@@ -202,6 +202,14 @@ If you are on Mac OS and using a compiler that supports OpenMP, you need to go t
|
||||
|
||||
in order to get the benefit of multi-threading.
|
||||
|
||||
Building with GPU support
|
||||
-------------------------
|
||||
If you want to build XGBoost4J that supports distributed GPU training, run
|
||||
|
||||
.. code-block:: bash
|
||||
|
||||
mvn -Duse.cuda=ON install
|
||||
|
||||
********
|
||||
Contents
|
||||
********
|
||||
|
||||
Reference in New Issue
Block a user