Release xgboost 1.2 with GPU support #133
Closed
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Description of changes:
This CR upgrades XGBoost to 1.2 and enables GPU support.
silent
parameter has been removed in 1.1 in favor ofverbose
.survival:aft
is added to support survival analysis: https://xgboost.readthedocs.io/en/latest/tutorials/aft_survival_analysis.html1.0-1-cpu-py3
.nvidia/cuda:${CUDA_VERSION}-base-ubuntu${UBUNTU_VERSION}
is a small image that contains a minimal set of CUDA runtime files.tree_method: gpu_hist
(and use the correct instance type, e.g.,p3.xlarge
,p3.2xlarge
) to enable GPU training.-cpu-py3
in the framework version is also redundant, and this CR proposes to drop the-<architecture>-<python version>
suffix. (However, we will keep the old tag format in the deployment pipelines for backwards compatibility. That is, we will tag the same image with two tags:1.2-1
and1.2-1-cpu-py3
.)Testing: tox, integration tests
By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.