|
- import GPUtil
- import json
- # 网址：https://github.com/anderskm/gputil
- GPUtil.showUtilization()
- #返回可用 GPU 的列表 ID。
- # order- 确定返回可用 GPU 设备 ID 的顺序。order应指定为以下字符串之一：
- # 'first'- 按升序排列可用的 GPU 设备 ID（默认）
- # 'last'- 按 id 降序排列可用的 GPU 设备 id
- # 'random'- 随机订购可用的 GPU 设备 ID
- # 'load'- 按负载递增排序可用的 GPU 设备 ID
- # 'memory'- 通过升序内存使用来排序可用的 GPU 设备 ID
- # limit- 将返回的 GPU 设备 ID 数量限制为指定数量。必须是正整数。（默认 = 1）
- # maxLoad- 被认为可用的 GPU 的最大当前相对负载。负载大于 的 GPUmaxLoad不会返回。（默认 = 0.5）
- # maxMemory- 被视为可用的 GPU 的最大当前相对内存使用量。maxMemory不返回当前内存使用量大于的 GPU 。（默认 = 0.5）
- # includeNan- 真/假标志，指示是否包括负载或内存使用为 NaN 的 GPU（指示无法检索使用情况）。（默认 = 假）
- # excludeID- ID 列表，应从可用 GPU 列表中排除。见GPU类描述。（默认 = []）
- # excludeUUIDexcludeID-除了它使用 UUID 之外，其他相同。（默认 = []）
- # 输出
- # deviceIDs - 所有可用 GPU 设备 ID 的列表。如果当前负载和内存使用量分别小于maxLoad和maxMemory，则认为 GPU 可用。该列表是根据 排序的order。返回的设备 ID 的最大数量由 限制limit。
- deviceIDs = GPUtil.getAvailable(order = 'memory', limit = 1, maxLoad = 0.8, maxMemory = 0.8, includeNan=False, excludeID=[], excludeUUID=[])
- print(deviceIDs)
- # 返回第一个可用的 GPU。可用性是根据当前的内存使用和负载确定的，排序由指定的顺序确定。如果找不到可用的 GPU，则会引发错误。使用默认值时，与getAvailable(order = 'first', limit = 1, maxLoad = 0.5, maxMemory = 0.5)
- # 输入
- # order- 见描述GPUtil.getAvailable(...)
- # maxLoad- 被认为可用的 GPU 的最大当前相对负载。负载大于 的 GPUmaxLoad不会返回。（默认 = 0.5）
- # maxMemory- 被视为可用的 GPU 的最大当前相对内存使用量。maxMemory不返回当前内存使用量大于的 GPU 。（默认 = 0.5）
- # attempts- 在放弃寻找可用 GPU 之前，该函数应进行的尝试次数。（默认 = 1）
- # interval- 每次尝试查找可用 GPU 的时间间隔（以秒为单位）。（默认 = 900 --> 15 分钟）
- # verbose- 如果True是，则在每次尝试之前打印尝试次数，如果找到可用的，则打印 GPU id。
- # includeNan- 见说明GPUtil.getAvailable(...)。（默认 = 假）
- # excludeID- 见说明GPUtil.getAvailable(...)。（默认 = []）
- # excludeUUID- 见说明GPUtil.getAvailable(...)。（默认 = []）
- # 输出
- # deviceID - 包含 1 个元素的列表，其中包含第一个可用的 GPU 设备 ID。如果当前负载和内存使用量分别小于maxLoad和maxMemory，则认为 GPU 可用。顺序和限制分别固定为'first'和1。
- deviceID = GPUtil.getFirstAvailable(order = 'first', maxLoad=0.5, maxMemory=0.5, attempts=1, interval=900, verbose=False)
- print(deviceID)
- GPUtil.showUtilization(all=True, attrList=None, useOldCode=True)
- # 打印所有 GPU 的当前状态（id、内存使用情况、uuid 负载）
- # 输入
- # all- 真/假标志，指示是否应显示 GPU 上的所有信息。覆盖attrList。
- # attrListGPU-要显示的属性列表列表。有关更多信息/示例，请参见代码。
- # useOldCode- 真/假标志，指示是否应使用显示 GPU 利用率的旧代码。
- # 输出
- # 没有任何
-
- GPUs = GPUtil.getGPUs()
- for gpu in GPUs:
- print(gpu.id)
- # 输入
- # 没有任何
- # 输出
- # GPUs- 所有 GPU 的列表。每个GPU对应于计算机中的一个 GPU，并包含一个设备 ID、相对负载和相对内存使用情况。
- if __name__ == "___main__":
- pass
|
