在Windows系统下,python脚本里面如何获取某个subprocess占用的GPU Memory和GPU利用率,CPU的Memory和CPU利用率
时间: 2024-02-03 18:12:16 浏览: 259
在 Windows 系统下,可以使用第三方库 `pynvml` 来获取 GPU Memory 和 GPU 利用率信息。安装该库可以使用以下命令:
```
pip install pynvml
```
然后在 Python 脚本中,可以按照以下方式获取 GPU Memory 和 GPU 利用率信息:
```python
import pynvml
# 初始化 pynvml
pynvml.nvmlInit()
# 获取第 0 个 GPU 的句柄
handle = pynvml.nvmlDeviceGetHandleByIndex(0)
# 获取 GPU Memory 信息
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print("GPU Memory Total: {} MB".format(meminfo.total / 1024 / 1024))
print("GPU Memory Used: {} MB".format(meminfo.used / 1024 / 1024))
# 获取 GPU 利用率信息
utilization = pynvml.nvmlDeviceGetUtilizationRates(handle)
print("GPU Utilization: {} %".format(utilization.gpu))
# 关闭 pynvml
pynvml.nvmlShutdown()
```
另外,可以使用 Python 自带的 `psutil` 库来获取 CPU Memory 和 CPU 利用率信息。安装该库可以使用以下命令:
```
pip install psutil
```
然后在 Python 脚本中,可以按照以下方式获取 CPU Memory 和 CPU 利用率信息:
```python
import psutil
# 获取 CPU Memory 信息
mem = psutil.virtual_memory()
print("CPU Memory Total: {} MB".format(mem.total / 1024 / 1024))
print("CPU Memory Used: {} MB".format(mem.used / 1024 / 1024))
# 获取 CPU 利用率信息
cpu_percent = psutil.cpu_percent(interval=1)
print("CPU Utilization: {} %".format(cpu_percent))
```
需要注意的是,`psutil.cpu_percent()` 的参数 `interval` 表示获取 CPU 利用率的时间间隔。在本例中,我们设置为 1 秒。
阅读全文