Как я могу использовать h2o4gpu в Google Colab? - PullRequest
0 голосов
/ 10 апреля 2020

Мне нужно обучить модель (Random Forest), используя scikit learn, и я хотел запустить ее в Google Colab, используя опцию GPU. Я знаю, что не могу напрямую использовать sklearn для этой цели, поэтому я установил h2o4gpu в Google Colab в качестве замены для замены, задав все параметры модели для использования с графическим процессором. Кажется, все работает нормально, но затем Google Colab говорит мне, что я на самом деле не использую графический процессор (он просит меня перейти на стандартную среду, поскольку я не использую графический процессор). К сожалению, я не могу обучить модель в стандартной среде, так как она требует слишком много ресурсов, и я получаю ошибку памяти.

Итак, мой вопрос: почему это не работает? Как правильно настроить h2o4gpu, чтобы я мог использовать опцию Google Colab GPU?

1 Ответ

0 голосов
/ 03 мая 2020

Привет, мой друг, я проверяю его и делаю гиты по этому поводу.

Установите h2o4gpu и tpot на google colab (GPU)

  1. export некоторые переменные env
  2. установить linux пакеты
  3. удалить sklearn и установить python пакеты
  4. наслаждайтесь быстрым автоматическим ML с gpu

Надеюсь, это поможет вам легче выполнять код.

...