python · pytorch | 限制程序仅使用 8 个线程

最近做 metaworld 实验时，发现 cpu 跑满了（做 dmcontrol 实验就没有这种情况，神奇）

可以限制程序仅使用 8 个线程。具体的，在 import torch 之后，添加

torch.set_num_threads(8)

该方法是师兄告诉我的。师兄说线程多了反而速度慢，一般来说一个实验用 8 个线程就行；有时候如果不限制，torch 会自动把 cpu 占满。