Skip to content

gpu support for dgx spark (and its partner devices) #11840

@busFred

Description

@busFred

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)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions