get_gpu_model.md

import os
import platform
import subprocess
import warnings
from distutils import spawn


def get_gpu_model_name() -> str:
    if platform.system() == "Windows":
        # If the platform is Windows and nvidia-smi
        # could not be found from the environment path,
        # try to find it from system drive with default installation path
        nvidia_smi = spawn.find_executable("nvidia-smi")
        if nvidia_smi is None:
            nvidia_smi = (
                    "%s\\Program Files\\NVIDIA Corporation\\NVSMI\\nvidia-smi.exe"
                    % os.environ["systemdrive"]
            )
    else:
        nvidia_smi = "nvidia-smi"

    try:
        p = subprocess.Popen(
            [nvidia_smi, "--query-gpu=name", "--format=csv,noheader"],
            stdout=subprocess.PIPE,
        )
        stdout, stderror = p.communicate()
        output = stdout.decode("UTF-8")
        lines = output.split(os.linesep)
        if len(lines) > 1:
            warnings.warn("Multiple GPUs found, can't determine benchmark target")
        gpu_name = lines[0]
    except OSError:
        gpu_name = "None"
    return gpu_name


if __name__ == '__main__':
    print(get_gpu_model_name())