如何在 H2O AutoML 中禁用 GPU答案

【问题标题】：How to disable GPUs in H2O AutoML如何在 H2O AutoML 中禁用 GPU
【发布时间】：2021-04-30 00:30:27
【问题描述】：

当我使用 H2O AutoML 进行实验时，我收到错误消息：“在抛出 'thrust::system::system_error' what() 的实例后调用终止：parallel_for failed: invalid resource handle”。这个错误信息comes fromXGBoost 是因为超出了GPU限制。

在使用常规 XGBoost 时，我将 cuda 可见设备参数设置为空白以禁用 GPU。然而，这个论点似乎在 H2O AutoML - XGBoost 实现中被忽略了。

import os
os.environ["CUDA_VISIBLE_DEVICES"] = ""

目前，H2O AutoML 中 gpu 上唯一的 xgboost can be run。

有人知道如何在 H2O AutoML 中禁用 GPU 吗？

作为一种解决方法，我暂时排除了 XGBoost 算法来运行我的实验。当我排除 XGBoost 但我不想放弃 XGBoost 的强大功能时，麻烦就过去了。

from h2o.automl import H2OAutoML
model = H2OAutoML(max_runtime_secs = 60*60*2, exclude_algos = ["XGBoost"])

【问题讨论】：

标签： gpu xgboost h2o automl h2o.ai

【解决方案1】：

这绝对是一个疏忽，我们需要添加打开/关闭和/或指定 GPU 的功能。我为此开了一个ticket。我想知道是否有办法同时在系统级别（H2O/Python 之外）暂时禁用 GPU？感谢您的报告！

【讨论】：

当然，可以在 docker 级别禁用 gpu，但如果您使用的是共享服务器，这可能是不可能的。正如我所提到的，我排除了 XGBoost，如果这是在 h2o 中处理的，我可以运行 AutoML，那就太棒了。谢谢。
我们会解决这个问题的。