Может ли кто-нибудь подробно описать шаги по сборке и установке XGBoost для нескольких графических процессоров?
Ниже приведен журнал нетривиального процесса сборки, раздел комментариев содержит правильные команды для скачать xgboost и субмодули. Я до сих пор не знаю, как установить, после сборки.
Мои попытки сделать эту работу привели меня на путь установки NCCL2 из nvidia , обновления cmake до версии 3.16. 4 (требуется> 3.12), и попытка собрать XGBoost для графического процессора .
Команды из документации XGBoost приведены ниже:
В каталоге клонированного Библиотека XGBoost от Github ,
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.