Как запустить прогноз на GPU?

Question

Я использую h2o4gpu и параметры, которые я установил

h2o4gpu.solvers.xgboost.RandomForestClassifier model.

XGBClassifier(base_score=0.5, booster='gbtree', colsample_bylevel=1,
colsample_bytree=1.0, gamma=0, learning_rate=0.1, max_delta_step=0,
max_depth=8, min_child_weight=1, missing=nan, n_estimators=100,
n_gpus=1, n_jobs=-1, nthread=None, num_parallel_tree=1, num_round=1,
objective='binary:logistic', predictor='gpu_predictor',
random_state=123, reg_alpha=0, reg_lambda=1, scale_pos_weight=1,
seed=None, silent=False, subsample=1.0, tree_method='gpu_hist')

Когда я тренирую эту модель, а затем прогнозирую, все работает нормально на GPU.

Тем не менее, когда я сохраняю свою модель в рассоле, а затем загружаю ее обратно в другой блокнот, а затем выполняю прогнозирование через предикат на нем, то все работает на процессоре.

Почему мой прогноз не работает на GPU?

Answer 1

Предсказания предназначены для запуска на процессоре, поэтому вам не нужен графический процессор для фактического использования модели.