Similar non-duplicate posts:
Edit xgboost/python-package/setup.py
and change line 38 to the following (source) :
include_package_data=False
Now it should install without any problems. To see that it's all working fine, just run the following command, and if it runs without errors it's good to go:
python -c "import xgboost"
You can run additional tests after installation using the nose
package with the following command executed from the root xgboost/
directory:
nosetests tests/python
To further confirm that it installed with GPU support, you can use the benchmarking scripts that come included with the installation:
gpu_hist algorithm:
python tests/benchmark/benchmark_tree.py
Output:
Train Time: 46.25219774246216 seconds
hist algorithm without GPU:
python tests/benchmark/benchmark_tree.py --tree_method hist
Output:
Train Time: 84.04853415489197 seconds