Linux 中如何获取 VRAM 和 GPU 缓存大小？

2024-03-11 23:41:02

Linux 中的 VRAM 和 GPU 缓存大小获取

简介

在 Linux 系统中，获取 VRAM 和 GPU 缓存大小对于监控系统资源和诊断图形问题至关重要。本文将详细介绍在没有 X11 或 Wayland 运行的情况下获取这些信息的可靠方法，以及跨平台解决方案和示例代码。

获取 VRAM 大小

对于 AMD GPU

lspci -vnn | grep -i 'VGA compatible controller' | grep -i 'memory at' | awk '{print $12}'

对于 NVIDIA GPU

lspci -vnn | grep -i 'VGA compatible controller' | grep -i 'memory at' | grep -i 'prefetchable' | awk '{print $12}'

获取 GPU 缓存大小

对于 AMD GPU

radeontop -d | grep "Vram cache" | awk '{print $7}'

对于 NVIDIA GPU

nvidia-smi --query-gpu=memory.l1.total --format=csv,noheader,nounits

解决可变大小 BAR 问题

可变大小 BAR（基址寄存器）会影响命令结果。解决方法有：

启用 BIOS 中的可变大小 BAR
使用 hwinfo 等第三方工具

跨平台解决方案

Windows： WMI 或 DirectX 诊断工具
macOS： IOKit 或 Apple System Information

示例代码（Python）

以下脚本结合上述命令，获取 VRAM 和 GPU 缓存大小：

import subprocess
import re

def get_vram_size():
    output = subprocess.check_output(["lspci", "-vnn"]).decode("utf-8")
    if "Advanced Micro Devices, Inc." in output:
        match = re.search("Memory at (.*) \\[size=(.*)\\]", output)
        return match.group(2) if match else None
    elif "NVIDIA Corporation" in output:
        match = re.search("Memory at (.*) \\[size=(.*)\\]", output)
        return match.group(2) if match else None
    else:
        raise Exception("Unknown GPU vendor")

def get_gpu_cache_size():
    output = subprocess.check_output(["radeontop", "-d"]).decode("utf-8")
    if "Advanced Micro Devices, Inc." in output:
        match = re.search("Vram cache: (.*)", output)
        return match.group(1) if match else None
    elif "NVIDIA Corporation" in output:
        import nvml
        nvml.init()
        handle = nvml.nvmlDeviceGetHandleByIndex(0)
        info = nvml.nvmlDeviceGetL1CacheInfo(handle)
        return info.size
    else:
        raise Exception("Unknown GPU vendor")

print("VRAM Size:", get_vram_size())
print("GPU Cache Size:", get_gpu_cache_size())