网上的配置方法参差不齐,踩了一些坑以后,记录一下自己配置成功的方法:
不管是window还是linux,cuda还是要装的,具体怎么装就不说了,下面主要说gpu版的xgboost的配置
windows:
跟着这篇博客的搞就行了。
linux(ubuntu):
跟着官方文档搞
跟着一步一步敲就行了,主要命令是:
git clone --recursive https://github.com/dmlc/xgboost
cd xgboost
从XGBoost目录开始:
mkdir build
cd build
cmake .. -DUSE_CUDA=ON
make -j4
cd python-package
sudo python setup.py install