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

【问题标题】：Can H2o AutoML benefit from a GPU instance on Sagemaker platform?H2o AutoML 能否从 Sagemaker 平台上的 GPU 实例中受益？
【发布时间】：2019-09-05 19:25:33
【问题描述】：

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

【问题讨论】：

您能否分享更多有关您如何使用 H2O AutoML 以及它是什么版本的 H2O 的信息？
@ErinLeDell，当然！我相信它是 H2O AutoML 的 3.2 版本。我是一名管理顾问，作为我工作的一部分，我使用机器学习来预测事件的可能性以及使用回归预测连续变量。此外，我一直在使用 ML 进行预测。具体来说，我在一个项目中，我需要预测投资组合在某个时间窗口内下跌的可能性。在这个项目中，我需要训练一个庞大的数据集，然后，我想我可以利用 GPU 处理来减少训练时间。通常，我在 Sagemaker 上使用笔记本。
您可能需要仔细检查版本号，我们现在是 3.26，所以它可能接近那个。
是的，@ErinLeDell 我意识到它是 3.26。

标签： gpu h2o amazon-sagemaker automl

【解决方案1】：

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

我的建议是在启用了 GPU 的实例上对其进行测试，并将结果与非 GPU 实例进行比较，看看是否值得额外花费。

【讨论】：

谢谢@ErinLeDell。我是否需要任何其他设置才能使 de GPU 实例与 H2O AutoML 一起使用？比如，安装一些库或其他什么？
您需要检查您是否安装了 Nvidia GPU 和 CUDA 8。您可以使用nvidia-smi 命令通过命令行监控 AutoML 运行期间的 GPU 使用情况。
谢谢@ErinLeDell，又是一个疑问！要同时安装两者，我只能安装 CUDA Toolkit 吗？我在Cuda的网站上看到CUDA的最新版本是10.1.243，我可以安装那个版本（10）还是我需要专门安装版本8？
@Erin LeDell，H2O AutoML XGBoost 能否在 Google Colab 中使用 TPU？
@Md.HishamurRahman 不，没有 TPU 支持，只有 GPU。