首页 > 解决方案 > H2o AutoML 能否从 Sagemaker 平台上的 GPU 实例中受益?

问题描述

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

标签: gpuh2oamazon-sagemakerautoml

解决方案


H2O AutoML包含一些算法,其中之一是XGBoost,它自 H2O 版本 3.22.0.1 以来一直是 H2O AutoML 的一部分。XGBoost 是 H2O AutoML 中唯一支持 GPU 的算法,但是,在 AutoML 中训练的许多模型都是 XGBoost 模型,因此使用 GPU 仍然很有用。请记住,您必须使用 H2O 3.22 或更高版本才能使用此功能。

我的建议是在启用了 GPU 的实例上对其进行测试,并将结果与​​非 GPU 实例进行比较,看看是否值得付出额外的成本。


推荐阅读