pytorch怎么在训练模型时查看CPU占用率
时间: 2023-05-25 20:04:29 浏览: 323
查看、显示CPU使用率
可以使用Python内置模块`psutil`来获取CPU占用率,具体代码如下:
```python
import psutil
for i in range(num_epochs):
# training code
cpu_percent = psutil.cpu_percent(interval=None)
print(f'Epoch {i}: CPU usage {cpu_percent}%')
```
其中`psutil.cpu_percent(interval=None)`返回当前时刻CPU占用率的百分比,`interval`参数可以设置查询时间间隔,默认为`None`表示立即查询。在每个epoch结束时查询当前CPU占用率并打印。
阅读全文