H2o AutoML能否从Sagemaker平台上的GPU实例中受益？

Question

我正在使用Sagemaker笔记本实例使用H2o AutoML运行一些项目，我想知道H2o AutoML是否可以从GPU Sagemaker实例中受益，如果可以，我应该如何配置笔记本？

Answer 1

H2O AutoML contains a handful of algorithms and one of them is XGBoost , which has been part of H2O AutoML since H2O version 3.22.0.1. XGBoost is the only GPU-capable algorithm inside of H2O AutoML, however, a lot of the models that are trained in AutoML are XGBoost models, so it still can be useful to utilize a GPU. Keep in mind that you must use H2O 3.22 or above to use this feature.

My suggestion is to test it on a GPU-enabled instance and compare the results to a non-GPU instance and see if it's worth the extra cost.