python查询gpu信息
时间: 2023-11-01 08:06:31 浏览: 38
可以使用第三方库 `nvidia-smi` 来查询 GPU 的信息。以下是使用 Python 调用 `nvidia-smi` 的示例代码:
```python
import subprocess
def get_gpu_info():
result = subprocess.run(['nvidia-smi', '--query-gpu=index,name,memory.total,memory.used,memory.free,temperature.gpu', '--format=csv'], stdout=subprocess.PIPE)
output = result.stdout.decode('utf-8')
gpu_info = output.strip().split('\n')[1:]
gpu_list = []
for info in gpu_info:
info = info.split(', ')
gpu_list.append({
'index': int(info[0]),
'name': info[1],
'memory_total': int(info[2].split()[0]),
'memory_used': int(info[3].split()[0]),
'memory_free': int(info[4].split()[0]),
'temperature': int(info[5].split()[0])
})
return gpu_list
print(get_gpu_info())
```
该函数将返回一个列表,其中包含每个 GPU 的信息,包括索引、名称、总内存、已用内存、空闲内存和温度。