cpu - 如何让 H2O 无人驾驶 AI 在 CPU 上使用更多内核？

Question

我的机器的 CPU 有 20 个核心，但是在运行 Driverless AI 时，它只使用了其中的 4 个。我怎样才能让它使用更多的核心以获得更快的结果？

Answer 1

这取决于您的设置和您使用的 DAI 版本，但您可以在config.toml文件中指定要使用的 CPU 数量。为了您的方便，我在下面粘贴了 toml 文件的相关部分，并提供了包含设置此文件的详细信息的文档链接。

## Hardware: Configure hardware settings here (GPUs, CPUs, Memory, etc.)

# Max number of CPU cores to use per experiment. Set to <= 0 to use all cores.
# One can also set environment variable "OMP_NUM_THREADS" to number of cores to use for OpenMP
# e.g. In bash: export OMP_NUM_THREADS=32 and export OPENBLAS_NUM_THREADS=32
#Set to -1 for all available cores.
#max_cores = -1

文档链接