解决PyTorch程序占用较高CPU

在使用服务器运行PyTorch程序时，使用TOP命令查看CPU状态，发现CPU占用率竟然达到2000%，已经影响到其他人正常使用服务器。

解决办法：
在main程序中加以下代码：

torch.set_num_threads(1)

设置最多使用1个CPU核。问题解决~