Skip to content

Commit

Permalink
Update xgboost requirements (#2898)
Browse files Browse the repository at this point in the history
Co-authored-by: Ziyue Xu <[email protected]>
  • Loading branch information
YuanTingHsieh and ZiyueXu77 authored Sep 3, 2024
1 parent 0bb9f6e commit 5ef22f0
Show file tree
Hide file tree
Showing 8 changed files with 15 additions and 23 deletions.
10 changes: 3 additions & 7 deletions examples/advanced/finance-end-to-end/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
pandas
# require xgboost 2.2 version, for now need to install a binary build
# "xgboost>=2.2"

--extra-index-url https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/list.html?prefix=federated-secure/
xgboost

scikit-learn
# require xgboost 2.2 version, for now need to install a nightly build
https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/federated-secure/xgboost-2.2.0.dev0%2B4601688195708f7c31fcceeb0e0ac735e7311e61-py3-none-manylinux_2_28_x86_64.whl
scikit-learn
3 changes: 1 addition & 2 deletions examples/advanced/finance/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,5 +5,4 @@ scikit-learn
torch
tensorboard
# require xgboost 2.2 version, for now need to install a nightly build
--extra-index-url https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/list.html?prefix=federated-secure/
xgboost
https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/federated-secure/xgboost-2.2.0.dev0%2B4601688195708f7c31fcceeb0e0ac735e7311e61-py3-none-manylinux_2_28_x86_64.whl
3 changes: 1 addition & 2 deletions examples/advanced/random_forest/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,5 +4,4 @@ scikit-learn
torch
tensorboard
# require xgboost 2.2 version, for now need to install a nightly build
--extra-index-url https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/list.html?prefix=federated-secure/
xgboost
https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/federated-secure/xgboost-2.2.0.dev0%2B4601688195708f7c31fcceeb0e0ac735e7311e61-py3-none-manylinux_2_28_x86_64.whl
3 changes: 1 addition & 2 deletions examples/advanced/vertical_xgboost/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,5 +4,4 @@ pandas
torch
tensorboard
# require xgboost 2.2 version, for now need to install a nightly build
--extra-index-url https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/list.html?prefix=federated-secure/
xgboost
https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/federated-secure/xgboost-2.2.0.dev0%2B4601688195708f7c31fcceeb0e0ac735e7311e61-py3-none-manylinux_2_28_x86_64.whl
3 changes: 1 addition & 2 deletions examples/advanced/xgboost/histogram-based/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,5 +4,4 @@ scikit-learn
torch
tensorboard
# require xgboost 2.2 version, for now need to install a nightly build
--extra-index-url https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/list.html?prefix=federated-secure/
xgboost
https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/federated-secure/xgboost-2.2.0.dev0%2B4601688195708f7c31fcceeb0e0ac735e7311e61-py3-none-manylinux_2_28_x86_64.whl
3 changes: 1 addition & 2 deletions examples/advanced/xgboost/tree-based/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,5 +4,4 @@ scikit-learn
torch
tensorboard
# require xgboost 2.2 version, for now need to install a nightly build
--extra-index-url https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/list.html?prefix=federated-secure/
xgboost
https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/federated-secure/xgboost-2.2.0.dev0%2B4601688195708f7c31fcceeb0e0ac735e7311e61-py3-none-manylinux_2_28_x86_64.whl
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
pandas
scikit-learn
xgboost
# require xgboost 2.2 version, for now need to install a nightly build
https://s3-us-west-2.amazonaws.com/xgboost-nightly-builds/federated-secure/xgboost-2.2.0.dev0%2B4601688195708f7c31fcceeb0e0ac735e7311e61-py3-none-manylinux_2_28_x86_64.whl
10 changes: 5 additions & 5 deletions integration/xgboost/encryption_plugins/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,12 @@


## Install required dependencies for building CUDA plugin
If you want to build the CUDA plugin on your own, you need to install the following libraries:
Require `libgmp3-dev`, CMake>=3.19, CUDA runtime >= 12.1, CUDA driver >= 12.1, NVIDIA GPU Driver >= 535
Compute Compatibility >= 7.0
If you want to build the CUDA plugin, you need to install the following libraries:
Require `libgmp3-dev`, CMake>=3.19, CUDA Driver and runtime 12.2 or 12.4, NVIDIA GPU Driver >= 535
Compute Compatibility >= 7.0.

On the building site:
1. Install GPU Driver >= 535, CUDA runtime >= 12.1, CUDA driver >= 12.1
1. Install GPU Driver >= 535, CUDA Driver and runtime 12.2 or 12.4
2. Install `libgmp3-dev`, gcc, CMake
3. Clone the NVFlare main branch and update the submodule
```
Expand Down Expand Up @@ -39,7 +39,7 @@ The generated plugin files under build folder are,
For each client site:

1. Copy the pre-built ".so" files to each site
2. Make sure you have installed required dependencies: GPU Driver >= 535, CUDA runtime >= 12.1, CUDA driver >= 12.1, `libgmp3-dev`
2. Make sure you have installed required dependencies: GPU Driver >= 535, CUDA Driver and runtime 12.2 or 12.4, `libgmp3-dev`
3. Update the site local resource.json to point to the appropriate ".so" file

For handling these complex environment, we recommend you build a docker image so every
Expand Down

0 comments on commit 5ef22f0

Please sign in to comment.