Как запустить xgboost на gpu
model = xgb.XGBRegressor(max_depth=i, n_estimators=j, tree_method = 'hist', device='CUDA')
Выдает ошибку:
[10:43:19] WARNING: C:\Users\dev-admin\croot2\xgboost-split_1675461376218\work\src\learner.cc:767: Parameters: { "device" } are not used.
Если прописать вот так:
model = xgb.XGBRegressor(max_depth=i, n_estimators=j, tree_method = 'gpu_hist')
То
c:\users\dev-admin\croot2\xgboost-split_1675461376218\work\src\common\common.h:239: XGBoost version not compiled with GPU support.
В чем собственно проблема? CUDA установлено, но в диспетчере задач не отображается:
Ответы (1 шт):
Ошибка вам выдаётся достаточно информативная.
XGBoost version not compiled with GPU support.
Установленная у вас версия библиотеки XGBoost скомпилирована без поддержки GPU. Вам нужно установить другую версию библиотеки - с поддержкой GPU. Это типичная ситуация для многих библиотек. Поддержка GPU сильно "раздувает" библиотеку, поэтому обычно выпускают отдельно версии библиотеки для CPU и для GPU.
Пример из документации установки GPU версии через conda:
conda install -c conda-forge py-xgboost-gpu