multithreading - Pytorch 使用了太多资源

Question

我正在使用 pytorch 来训练 DQN 模型。使用 ubuntu，如果我使用 htop，我会得到

正如你所看到的，所有的资源都被使用了，我有点担心。这是我的代码。

有没有办法使用更少的资源？我必须使用 pytorch 添加我的要求吗？

请注意，我的机器上没有 GPU，只有 CPU

Answer 1

就在这里。您可以使用torch.set_num_threads(...)来指定线程数。根据您使用的 PyTorch 版本，此功能可能无法正常工作。在本期中了解原因。在那里，您会看到，如果需要，您可以使用环境变量来分别通过OMP_NUM_THREADS=?和来限制 OpenMP 或 MKL 线程的使用MKL_NUM_THREADS=?，其中?是线程数。

请记住，这些东西预计将在具有数千个内核的 GPU 上运行，因此我只会在极其必要时限制 CPU 使用率。