PyTorch 限制 CPU 线程数目

终端命令 htop ：查看 CPU 利用率

限制 python 的cpu 利用率：

import os
cpu_num = 4 # 这里设置成你想运行的CPU个数
os.environ["OMP_NUM_THREADS"] = str(cpu_num)  # noqa
os.environ["MKL_NUM_THREADS"] = str(cpu_num) # noqa
torch.set_num_threads(cpu_num )

原因

1. tensor 没有 .to('cuda') ,tensor 还是部署在 cpu 上，耗的是 cpu 的资源，导致 cpu 占用率过高。
2. 在读数据的时候设置 num_workers = 0，也可能会导致 cpu 占用率过高。