如何查看服务器有几张GPU

病树前头2025-05-31 10:06

要查看服务器上有多少张 GPU，你可以使用以下几种方法：

如果你的服务器上安装了 NVIDIA GPU 驱动程序，那么可以使用 nvidia-smi 命令查看详细的 GPU 信息。

bash 复制代码

nvidia-smi

该命令将显示服务器上所有的 NVIDIA GPU 及其详细信息，包括驱动版本、每个 GPU 的使用情况、内存使用、温度等。

这个命令可以用于列出所有的 PCI 设备，包括 GPU。在命令输出中，你可以过滤出与显卡相关的信息。

bash 复制代码

lspci | grep -i --color 'vga\|3d\|2d'

这将显示所有与图形相关的设备，不过它可能不会显示 GPU 的具体型号和详细信息。

lshw 是一个可以列出硬件的详细信息的工具。你可以用它来列出所有图形设备。

bash 复制代码

sudo lshw -C display

这将提供显示控制器的详细信息。如果没有显示出 GPU 型号，确保 lshw 已正确安装，并以 sudo 权限运行。

glxinfo 是一个用于显示 OpenGL 和图形相关信息的工具。

bash 复制代码

glxinfo | grep "OpenGL renderer"

这种方法也提供当前机器上可用的 GPU 信息，但其主要用于图形界面的环境。

在 Python 中，使用 PyCUDA 库：

python 复制代码

import pycuda.driver as cuda
cuda.init()
print(f"Detected {cuda.Device.count()} CUDA-capable device(s)")

使用 TensorFlow（如果 TensorFlow 和 CUDA 已正确安装）：

python 复制代码

import tensorflow as tf
gpus = tf.config.list_physical_devices('GPU')
print(f"Available GPU(s): {len(gpus)}")

通过这些方法，你应该能够确定服务器上安装了多少 GPU，并获取一些基本的信息。如果正在使用的是其他品牌的 GPU（比如 AMD），则需要相应的工具来获取信息。