automl CLI on GPU

Question

Hi,

I am building a software that is calling automl classification and automl regression from C# on windows. I have a strong GPU and I would like to utilize it for the training process.

Does automl supports GPU utilization? I saw that it is somehow possible through this but it seems to work only from the UI based model builder on visual studio.

Is it possible to make it work from the automl CLI or powershell based tool? and will it work on regular machine learning (not deep learning)?

Thanks :)

Answer

Hi @super richmann - Do you mean ML.NET AutoML (C#) or Azure AutoML.

The only case where ML.NET AutoML (C#) can use GPU is when training Image related models (such as ImageClassification), I believe.
Most ML.NET algorithms are classical algorithms and would not get much improvement if using GPU (Probably LightGBM, but I don't think GPU is supported for classical algorithms in ML.NET).

The second (Azure AutoML) is Python based and if using Azure ML compute you can use GPU for DNN operations such as text featurization using BERT under the covers. For Azure AutoML we're also releasing CLI and REST API to trigger training jobs in the Azure cloud.

I work for the Azure AutoML team, so if you are interested on enrolling the CLI/REST-API private preview, send me an email to cesardl at microsoft.com, ok?
Azure AutoML Python SDK is GA for more than 1 year, though (without CLI and REST API).

Share via

automl CLI on GPU

1 answer

Your answer