python查看linux有无gpu

在Python中，你可以使用一些第三方库来检查Linux系统上是否存在GPU。其中一个常用的库是`psutil`，它主要用于系统监控，包括CPU、内存、磁盘等信息，但对于直接检测GPU，可能不是最准确的选择。但是，你可以尝试通过它间接获取相关信息。

另一种方法是使用`nvidia-smi`命令行工具，这个是NVIDIA提供的用于管理系统显卡状态的实用程序。在Python中，可以利用os模块的Popen函数执行命令并读取结果，例如：

import os

def check_gpu():
    try:
        output = subprocess.check_output(['nvidia-smi', '--query-gpu=driver_version,memory.total'], text=True)
        if 'Driver Version' in output and 'Total Memory' in output:
            return True
    except FileNotFoundError:
        pass
    return False

if check_gpu():
    print("系统上有GPU")
else:
    print("系统上没有GPU")

#