linux下安装XGBoost并支持GPU(anaconda3)

这几天为了让xgboost运行的更快,所以在研究让他支持gpu版本。踩了不少的坑,在此贡献出来让大家少点烦恼。

前提

     笔者由于开始是小菜鸟一枚,系统里直接安装的anaconda3,因此之前装的cpu版的xgboost是参考网上给出的直接利用命令行来安装:pip install xgboost

    但是,如果直接按照官网上建议的GPU支持指南操作会发现,xgboost文件夹下已经存在build文件夹,不需要自己创建,如果直接cd进入build文件夹,然后执行

cmake .. -DUSE_CUDA=ON

将会提示没有Cmake.list(大致是这个文件,具体形式记不清了),因此,为了能够使用gpu小编就使用:pip uninstall xgboost卸载了它,按照官网配置重来一遍。

 心酸历程

    首先,需要将cmake更新到3.2以上,更新方法可参考cmake更新教程。然后就可以按照官网上给出的方案进行配置了,但是。。。。。。一定要注意的一点是:如果要支持gpu,执行完

git clone --recursive https://github.com/dmlc/xgboost
cd xgboost

    不能执行后面的 make -j4!!!如果执行就会自动出现build,后面一切支持gpu的操作全部白费。这里正确的操作是直接跳到linux支持gpu的章节上直接执行

$ mkdir build
$ cd build
$ cmake .. -DUSE_CUDA=ON
$ make -j
    这样就基本完成了xgboost的安装以及对gpu的支持,最后按照官网提示进行python package的安装,即可调用xgboost的gpu尽行加速操作了。最后的最后贴出安装官网 点击打开链接,网上的东西真的有太多太多坑,官网上的菜最靠谱,希望大家都慢慢试着直接看官网学习知识。

猜你喜欢

转载自blog.csdn.net/wl2858623940/article/details/80546140