gpu - 如何在 H2O AutoML 中禁用 GPU

Question

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

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

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

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

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

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

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

Answer 1