Использование LightGBM в режимах CPU и GPU

Question

В настоящее время я использую LightGBM в режиме CPU и хочу попробовать его в режиме GPU. Флаги для режимов CPU и GPU различны. Какие есть варианты, чтобы я мог переключаться между режимами CPU и GPU? Является ли использование conda env единственным вариантом?

Answer 1

Проверьте документы здесь https://lightgbm.readthedocs.io/en/latest/GPU-Tutorial.html?highlight=gpu#run-your-first-learning-task-on-gpu , а также на странице GitHub: https://github.com/Microsoft/LightGBM/blob/master/docs/Installation-Guide.rst