【Python】获取nvidia-smi中gpu相关的信息

最新推荐文章于 2024-07-26 10:08:40 发布

拓荒探险者

最新推荐文章于 2024-07-26 10:08:40 发布

阅读量2.2k

点赞数 2

分类专栏： # Python 文章标签： python nvidia gpu 显卡

本文链接：https://blog.csdn.net/zxc120389574/article/details/106220612

版权

本文介绍如何利用Python解析`nvidia-smi`命令输出，获取GPU的相关信息，包括显卡型号、内存使用等。

摘要由CSDN通过智能技术生成

import pynvml
import time
pynvml.nvmlInit()

def printNvidiaGPU(gpu_id):
    # get GPU temperature
    gpu_device = pynvml.nvmlDeviceGetHandleByIndex(gpu_id)

    temperature = pynvml.nvmlDeviceGetTemperature(gpu_device, pynvml.NVML_TEMPERATURE_GPU)
    # get GPU memory total
    totalMemory = pynvml.nvmlDeviceGetMemoryInfo(gpu_device).total
    # get GPU memory used
    usedMemory = pynvml.<