XGBoost GPU regression fails at predict time with Check failed: dmat->Device() when training with tree_method=’hist’ and device=’cuda’
Summary The XGBoost GPU regression fails at predict time with a Check failed: dmat->Device() error when training with tree_method=’hist’ and device=’cuda’. This issue arises due to a device/context mismatch between the training and prediction phases. The error occurs when the input at prediction time is a NumPy array versus a pandas DataFrame, or when switching … Read more