时间戳

首页 > 解决方案 > 如何在 Ubuntu 18.04 工作站上为多 GPU 构建 XGBoost

cmake - 如何在 Ubuntu 18.04 工作站上为多 GPU 构建 XGBoost

问题描述

谁能详细描述为多 GPU 构建和安装 XGBoost 的步骤？

下面是非平凡构建过程的日志，评论部分包含下载 xgboost 和子模块的正确命令。我仍然不清楚安装过程，构建后。

我试图完成这项工作使我走上了从 nvidia 安装 NCCL2的道路，将 cmake 升级到版本 3.16.4（需要> 3.12），并尝试为 GPU 构建 XGBoost。

XGBoost 文档中的命令总结如下：

在从 Github 克隆的 XGBoost 库的目录中，

mkdir build
cd build
cmake … -DUSE_CUDA=ON -DUSE_NCCL=ON -DNCCL_ROOT=/path/to/nccl2
make -j4

最后，如果“make”命令有效，你如何安装这个版本？文档指的是python3 setup.py install在 xgboost/python-package 文件夹中安装，但是这个链接是指向刚刚创建的构建文件夹还是没有多 GPU 支持的主构建？

我的工作站运行的是 Ubuntu 18.04 LTS、CUDA 10.2、nvidia 驱动程序 440.44。

标签： cmakegpuxgboost

解决方案

推荐阅读