Support GPU training in the NVFlare demo (#7965)

This commit is contained in:
Rong Ou
2022-06-02 06:52:36 -07:00
committed by GitHub
parent 6b55150e80
commit 31e6902e43
4 changed files with 26 additions and 3 deletions

View File

@@ -20,7 +20,12 @@ Build the Plugin
# Under xgboost source tree.
mkdir build
cd build
cmake .. -GNinja -DPLUGIN_FEDERATED=ON
# For now NCCL needs to be turned off.
cmake .. -GNinja\
-DPLUGIN_FEDERATED=ON\
-DUSE_CUDA=ON\
-DBUILD_WITH_CUDA_CUB=ON\
-DUSE_NCCL=OFF
ninja
cd ../python-package
pip install -e . # or equivalently python setup.py develop
@@ -31,5 +36,6 @@ Test Federated XGBoost
```shell
# Under xgboost source tree.
cd tests/distributed
# This tests both CPU training (`hist`) and GPU training (`gpu_hist`).
./runtests-federated.sh
```