终端命令 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 )
原因
- tensor 没有
.to('cuda')
,tensor 还是部署在 cpu 上,耗的是 cpu 的资源,导致 cpu 占用率过高。 - 在读数据的时候设置 num_workers = 0,也可能会导致 cpu 占用率过高。