python - 如何限制 GluonCV 使用的核心数量

Question

我正在尝试相同的 GluonCV 推理示例。但是我想在单个核心上对结果进行基准测试。我尝试使用 ctx 给 cpu(0) 但仍然无法限制正在使用的内核数量。

我尝试了以下方法，但这将设置为我想设置的完整脚本，仅用于推理 import os os.environ["OMP_NUM_THREADS"] = 1 # export OMP_NUM_THREADS=1

标签： pythonmxnet