-
-
Notifications
You must be signed in to change notification settings - Fork 8.8k
Description
I'm on Lenovo ThinkStation PGX (NVidia DGX Spark's Lenovo equivalent); I have xgboost.__version__ = '3.1.2' installed via pip install xgboost. I tried device="cuda" and device="gpu" and both yield the same warning and fallback to cpu.
import xgboost as xgbst
import torch as th
x = th.randn((279, 580))
y = th.randn((279, 1))
xgbr = xgbst.XGBRegressor(device="cuda")
xgbr.fit(x.numpy(), y.numpy())/home/hungtien/miniconda3/envs/tmplafa/lib/python3.13/site-packages/xgboost/training.py:199: UserWarning: [13:55:49] WARNING: /workspace/src/context.cc:207: Device is changed from GPU to CPU as we couldn't find any available GPU on the system.
bst.update(dtrain, iteration=i, fobj=obj)
xgbr = xgbst.XGBRegressor(device="gpu")
xgbr.fit(x.numpy(), y.numpy())/home/hungtien/miniconda3/envs/tmplafa/lib/python3.13/site-packages/xgboost/training.py:199: UserWarning: [13:59:45] WARNING: /workspace/src/context.cc:207: Device is changed from GPU to CPU as we couldn't find any available GPU on the system.
bst.update(dtrain, iteration=i, fobj=obj)