如何在 H2O AutoML 中禁用 GPU

Question

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

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

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

目前，唯一可以在 H2O AutoML 中的 gpu 上运行xgboost。

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

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

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

Answer 1

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