[发明专利]一种GPU的设备信息获取方法及装置

说明书

本申请提供了一种GPU的设备信息获取方法，所述方法包括：分别从系统层和物理层获取GPU设备数量及GPU的ID；判断从系统层获取的GPU设备数量及GPU的ID与从物理层获取的GPU设备数量及GPU的ID是否相同；判断相同时，分别从所述系统层和所述物理层获取GPU的设备信息。利用本申请提供了一种GPU的设备信息获取方法，操作简单，实用性强，能够减少了人工操作及干预的时间。本申请还提供了一种GPU的设备信息获取装置。

技术领域

本发明涉及计算机技术领域，尤其涉及一种GPU的设备信息获取方法及装置。

背景技术

目前随着AI技术应用越来越广泛,带有GPU设备的服务器产品是不可缺少的重要硬件部分,用于图形、图像处理领域的GPU可以并行处理大量数据，非常适合深度学习的高并行、高本地化数据场景，是目前主流的AI人工智能计算架构。

NVIDIA的GPU目前在服务器产品中占主导地位,在实际设计中,GPU的数量动辄8个，16个，未来设计可能会更多的数量，针对此状况服务器产品开发过程中对于GPU信息抓取比对暂时没有一个有效直观的工具及方法，依靠人为的进行每个GPU设备的信息比对效率比较低而且出现遗漏错误。而且目前开发过程中已经有出现系统下GPU数量在不同环境下信息不同的异常发生。

发明内容

为了解决现有技术存在的上述技术问题，本申请提供了一种GPU的设备信息获取方法及装置，应用于服务器GPU设备的数量及信息抓取,操作简单，实用性强，能够减少了人工操作及干预的时间。

本申请提供了一种GPU设备的信息获取方法，所述方法包括：

分别从系统层和物理层获取GPU设备数量及GPU的ID；

判断从系统层获取的GPU设备数量及GPU的ID与从物理层获取的GPU设备数量及GPU的ID是否相同；

判断相同时，分别从所述系统层和所述物理层获取GPU的设备信息。

可选的，若从所述系统层获取的GPU设备数量及GPU的ID与从所述物理层获取的GPU设备数量及GPU的ID不同，所述方法还包括：

中断进程，并及时的进行错误提示。

可选的，所述GPU的设备信息至少包括以下其中一种：

GPU设备的带宽速率、GPU数量和显卡基本输入输出系统VBIOS。

可选的，所述分别从所述系统层和所述物理层获取GPU的设备信息后，所述方法还包括：

将从所述系统层和所述物理层获取的所述GPU的设备信息，分别保存到不同的日志文件中。

可选的，所述方法还包括，对所述GPU的ID进行排序处理。

本申请实施例还提供了一种GPU设备的信息获取装置，所述装置包括：第一获取模块、判断模块和第二获取模块；

所述第一获取模块，用于分别从系统层和物理层获取GPU设备数量及GPU的ID；

所述判断模块，用于判断从系统层获取的GPU设备数量及GPU的ID与从物理层获取的GPU设备数量及GPU的ID是否相同；

所述第二获取模块，用于判断相同时，分别从所述系统层和所述物理层获取GPU的设备信息。

可选的，其特征在于，若从所述系统层获取的GPU设备数量及GPU的ID与从所述物理层获取的GPU设备数量及GPU的ID不同，所述判断模块，还用于：

中断进程，并及时的进行错误提示。

可选的，其特征在于，所述GPU的设备信息至少包括以下其中一种：