首页 > 解决方案 > Docker 上的 H2O 多核初始化

问题描述

我正在尝试使用以下命令通过 Python 3.6.8 启动 H2O 3.26.0.3:

h2o.init(nthreads=-1)

在 16 核虚拟机上,H2O 使用以下设置正确初始化:

H2O cluster total cores:    16
H2O cluster allowed cores:  16

但在 16 核 Docker 容器上,H2O 使用以下不寻常的设置进行初始化:

H2O cluster total cores:    1
H2O cluster allowed cores:  1

H2O 无法识别容器设置中的所有 CPU 内核是否有原因?

编辑 如果我修改初始化以明确列出所有核心:

h2o.init(nthreads=16)

然后设置如下所示:

H2O cluster total cores:    1
H2O cluster allowed cores:  16

允许的核心数可以超过总核心数似乎很奇怪。但经过一些测试,H2O 似乎确实使用了多核而不是单核。因此,即使 H2O 没有检测到多个内核,它仍然可以利用它们。

标签: pythondockerh2o

解决方案


推荐阅读