How to enable GPU in XGBoost on apple M1 Pro chipset

126 Views Asked by At

I tried using GPU on XGBoost training on windows with device = cuda it worked and training time reduced drastically now i want to do this experiment on my Mac M1 Pro.

How to enable GPU for XGBoost on m1 pro chip set.

I tried looking into documentation where i was unable to find information.

1

There are 1 best solutions below

1
Itamar Trainin On

Currently XGBoost doesn't support MPS which is Mac's internal "GPU". To use GPU acceleration, you need to get CUDA gpus.

XGBoost Supported Devices