multithreading - Pytorch 使用了太多资源

Question

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



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

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

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

Answer 1

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

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