如何在谷歌Colab上使用h2o4gpu？

Question

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

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

Answer 1

嗨，我的朋友，我检查了它，并对它做了git。

Install h2o4gpu and tpot on google colab (GPU)

python variable

• install导出一些环境linux packges

• uninstall sklearn和安装包

1. 使用

享受fast auto ML

我希望这能帮助你更容易地运行你的代码。