我需要使用scikit learn训练一个模型(随机森林(,我想使用GPU选项在Google Colab上运行它。我知道我不能直接使用sklearn,所以我在Google Colab上安装了h204gpu作为替代品,设置了模型的所有参数,使其与GPU一起使用。一切似乎都很好,但谷歌Colab告诉我,我实际上并没有使用GPU(它要求我切换到标准环境,因为我没有使用GPU(。不幸的是,我无法在标准环境中训练模型,因为它需要太多的资源,而且我遇到了内存错误。
所以,我的问题是,为什么它不起作用?如何正确设置h204gpu,以便使用Google Colab GPU选项?
嗨,我的朋友,我检查了一下,并做了gits。
在谷歌colab(GPU(上安装h204gpu和tpot
- 导出一些env变量
- 安装linux包
- 卸载sklearn并安装python包
- 使用gpu享受快速自动ML
我希望这能帮助您更轻松地运行代码。