Pytorch 使用了太多资源

Question

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

如您所见，所有资源都已使用，我有点担心。 这是我的code。

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

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

Answer 1

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

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