sklearn-onnx icon indicating copy to clipboard operation
sklearn-onnx copied to clipboard

Published 20 hours ago • verified publisher icon onnx

Can onnx converted by "convert_sklearn" run on GPU?

Open GoHard2023 opened this issue 2 years ago • 2 comments

As the title, If I convert RandomForestClassifier to an ONNX model, how can we run it on GPUs?

GoHard2023 avatar Aug 27 '22 17:08 GoHard2023

It depends on the runtime you are using. onnxruntime has currently only one cpu implementation for TreeEnsembleClassifier.

xadupre avatar Aug 29 '22 08:08 xadupre

You could use hummingbird to convert your tree into a neural network and then run it on GPU.

xadupre avatar Oct 07 '22 14:10 xadupre