python - 如何在 Google Colab 上使用 h2o4gpu？

Question

我需要使用 scikit learn 训练一个模型（随机森林），并且我想使用 GPU 选项在 Google Colab 上运行它。我知道我不能为此直接使用 sklearn，所以我在 Google Colab 上安装了 h2o4gpu 作为替代品，设置模型的所有参数以将其与 GPU 一起使用。一切似乎都运行良好，但是 Google Colab 告诉我我实际上并没有使用 GPU（它要求我切换到标准环境，因为我没有使用 GPU）。不幸的是，我无法在标准环境中训练模型，因为它需要太多资源并且出现内存错误。

所以，我的问题是，为什么它不起作用？如何正确设置 h2o4gpu 以便可以使用 Google Colab GPU 选项？

Answer 1

嗨，我的朋友，我检查了它并对此进行了讨论。

在 google colab (GPU) 上安装 h2o4gpu 和 tpot

1. 导出一些环境变量
2. 安装 linux 包
3. 卸载sklearn并安装python包
4. 使用 gpu 享受快速的自动机器学习

我希望这可以帮助您更轻松地运行代码。