是h2o-xgboost支持的多CPU?

时间:2017-10-02 10:42:35

标签: h2o

是否有配置允许在多线程中运行 H2OXGBoostEstimator 而不是在具有一个CPU的最小配置中运行h2o版本3.15.0.4035?

1 个答案:

答案 0 :(得分:1)

H2O上的xgboost实现是多线程的,并且像H2O中支持的所有其他算法一样,但它依赖于平台,并且正确地描述为H2O documentation

因此,如果您在Linux上尝试它,并且所有支持的库都可用,那么您将利用分布式xgboost,否则就像OSX一样,您可能会获得单个CPU回退运行时。所以这完全取决于从您的操作系统加载哪个lib。

当H2O在日志中启动时,您将看到以下内容:

10-02 09:25:34.579 10.0.0.46:54321       54229  main      INFO: Registered 3 core extensions in: 57ms
10-02 09:25:34.580 10.0.0.46:54321       54229  main      INFO: Registered H2O core extensions: [Watchdog, XGBoost, KrbStandalone]
10-02 09:25:34.791 10.0.0.46:54321       54229  main      INFO: Registered: 161 REST APIs in: 211ms
10-02 09:25:34.791 10.0.0.46:54321       54229  main      INFO: Registered REST API extensions: [XGBoost, Algos, AutoML, Core V3, Core V4]

然后您将看到是否包含CPU / GPU,如下所示:

10-02 09:23:49.952 10.0.0.46:54321       54143  FJ-1-5    INFO: No GPU (gpu_id: 0) found. Using CPU backend.

如果您可以运行objdump或ldd命令来查看使用H2O加载的库,那么您将更好地了解导致xgboost运行时为单CPU的错误。