Linux下安装Xgboost并支持GPU
下载源码
git clone --recursive https://github.com/dmlc/xgboost
#若安装CPU版,直接执行以下命令
cd xgboost
make -j4
支持GPU
正常流程
$ cd xgboost
$ mkdir build
$ cd build
$ cmake .. -DUSE_CUDA=ON
切记:不要返回上一层目录(不要执行cd …)
直接执行:
$ make -j4
报错:nvcc fatal : redefinition of argument ‘std’
原因:编译参数中重复出现 -std=c++11
上网搜索的主要原因是:cmake版本新旧导致。
查看本机cmake版本
$ cmake --version
cmake version 3.17.0-rc2
CMake suite maintained and supported by Kitware (kitware.com