Python：获取当前电脑中空余容量最大的GPU

本文主要是介绍Python：获取当前电脑中空余容量最大的GPU，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

import pynvmlpynvml.nvmlInit()  # 初始化
deviceCount = pynvml.nvmlDeviceGetCount()  # 获得计算机中GPU的个数
largest_free_mem = 0
largest_free_idx = 0
for i in range(deviceCount):handle = pynvml.nvmlDeviceGetHandleByIndex(i)info = pynvml.nvmlDeviceGetMemoryInfo(handle)if info.free > largest_free_mem:largest_free_mem = info.freelargest_free_idx = i
pynvml.nvmlShutdown()
largest_free_mem = largest_free_mem / 1024. / 1024.  # Convert to MBidx_to_gpu_id = {}
for i in range(deviceCount):idx_to_gpu_id[i] = '{}'.format(i)  # 获得所有gpu的IDgpu_id = idx_to_gpu_id[largest_free_idx]
print(gpu_id, largest_free_mem)

这篇关于Python：获取当前电脑中空余容量最大的GPU的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---