使用 python 获取硬件信息
本文最后更新于 2024年4月26日 凌晨
1. 获取内存信息
1.1. 使用 psutil 获取内存信息
psutil 为 python 内置组件,不需要安装
可以应用于获取实时内存占用等场景
引用 psutil
1
import psutil
获取物理内存信息
1
2
3virtual_memory_status = psutil.virtual_memory()
# 转为字典格式
virtual_memory_status = psutil.virtual_memory()._asdict()获取交换内存信息
1
2
3swap_memory_status = psutil.virtual_memory()
# 转为字典格式
swap_memory_status = psutil.virtual_memory()._asdict()
2. 获取 CPU 信息
2.1. 使用 py-cpuinfo 获取 CPU 信息
安装 py-cpuinfo
1
pip install py-cpuinfo
引用 py-cpuinfo
1
import cpuinfo
获取 CPU 信息
1
cpu_info = cpuinfo.get_cpu_info()
2.2. 使用 psutil 获取 CPU 使用率
psutil 为 python 内置组件,不需要安装
可以应用于获取实时 CPU 使用率等场景
引用 psutil
1
import psutil
获取 CPU 使用率
1
cpu_percent = psutil.cpu_percent()
3. 获取 GPU 信息
3.1. 使用 gpustat 获取 GPU 信息
可以应用于实时获取 显存 占用等场景
安装 gpustat
1
pip install gpustat
引用 gpustat
1
import gpustat
获取 gpu 信息
1
gpu_info = gpustat.GPUStatCollection.new_query().jsonify()['gpus'][0]
输出参考
1 |
|
3.2. 使用 numba 获取 NVIDIA GPU 名称
安装 numba
1
pip install numba
从 numba 引用 cuda
1
from numba import cuda
判断是否有 NVIDIA GPU
1
cuda.is_available()
获取 NVIDIA GPU 名称
1
gpu_name = cuda.get_current_device()
使用 python 获取硬件信息
https://blog.cc01cc.cn/2023/03/16/python-hardware-info/