Ubuntu服务器安装XGBoost和LightGBM的GPU版本

在没有权限的账号上安装cmake安装XGBoost的GPU版本安装LightGBM的GPU版本

在没有权限的账号上安装cmake

如果有cmake可以跳过此步 首先下载cmake源码

cd cmake-3.15.2 ./bootstrap ./configure --prefix=/xxx/xxx/xxx

上面的prefix为当前账户下想要安装cmake的文件夹的完整目录

make make install

安装完成后将cmake添加到当前用户的环境变量中

vi ~/.bashrc

在下面添加

export PATH="/xxx/xxx/xxx/bin:$PATH"

保存后

source ~./bashrc

可以使用cmake --version测试是否安装好

安装XGBoost的GPU版本

首先下载xgboost源码，然后使用cmake编译

git clone --recursive https://github.com/dmlc/xgboost cd xgboost mkdir build cd build cmake .. -DUSE_CUDA=ON make -j

接下来将编译好的xgboost安装到python中

cd ../python-package python setup.py install

安装LightGBM的GPU版本

首先安装依赖库

sudo apt-get install --no-install-recommends git cmake build-essential libboost-dev libboost-system-dev libboost-filesystem-dev

使用pip安装LightGBM的GPU版本

pip install lightgbm --install-option=--gpu --install-option="--opencl-include-dir=/usr/local/cuda/include/" --install-option="--opencl-library=/usr/local/cuda/lib64/libOpenCL.so"