如何/在哪里将 Nvidia nccl2 安装到特定目录,以便运行支持多 GPU 的 XGBoost?

如何/在哪里将 Nvidia nccl2 安装到特定目录,以便运行支持多 GPU 的 XGBoost?

我需要将 Nvidia 的 nccl2 库安装到特定目录,以便运行支持 GPU 的 XGBoost。如何在指定要构建/保存的目录的同时安装 nccl 库,以及适当的目录或默认目录在哪里?

该库可以从 nvidia 下载为 .deb 文件这里您需要 Nvidia 开发者会员资格。Github 是这里

XGBooost 文档有构建说明:

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

所以我需要提前知道 nccl2 安装的路径(不清楚他们指的是 nccl.h 文件还是文件夹)。

相关内容