sparkling-water icon indicating copy to clipboard operation
sparkling-water copied to clipboard

Published 20 hours ago • verified publisher icon h2oai

multi-gpu support

Open dataspekulant opened this issue 4 years ago • 2 comments

I would like to ask if there is possibility to utilize more than 1 GPU during XGBoost training? If I understand correctly, gpu_id parameter accepts single value only, not a list.

Running PySparkling (3.30.1.1-1-2.4) on YARN.

dataspekulant avatar Oct 13 '20 18:10 dataspekulant

cc @honzasterba @michalkurka

mn-mikke avatar Oct 14 '20 10:10 mn-mikke

Hi @dataspekulant, This isn't currently supported. However, there is a plan to change it, you can track the progress of https://0xdata.atlassian.net/browse/PUBDEV-7843.

mn-mikke avatar Oct 19 '20 11:10 mn-mikke