Can I run CatBoost/XGBoost on my GPU(s) on my Mac?

I'm interested in using CatBoost and XGBoost for some machine learning projects on my Mac, and I was wondering if it's possible to run these algorithms on my GPU(s) to speed up training times.

I have a Mac with an AMD Radeon Pro 5600M and an Intel UHD Graphics 630 GPUs, and I'm running macOS Ventura 13.2.1. I've read that both CatBoost and XGBoost support GPU acceleration, but I'm not sure if this is possible on my system.

Can anyone point me in the right direction for getting started with GPU-accelerated CatBoost/XGBoost on macOS? Are there any specific drivers or tools I need to install, or any other considerations I should be aware of? Thank you.

Up vote post of Yesterdays Down vote post of Yesterdays
1.7k views
Posted by
Yesterdays
Reply
Add a Comment

Replies

In CatBoost you can use the "task_type" property, for example, like this CatBoostRegressor(task_type="GPU"), and that's it. See the documentation link: https://catboost.ai/en/docs/features/training-on-gpu.

Posted by
ArtemIsak
Up vote reply of ArtemIsak Down vote reply of ArtemIsak
Add a Comment