GPUtil GPUtil是一个Python模块,用于使用nvidia-smi从NVIDA GPU获取GPU状态。
GPUtil查找计算机上的所有GPU,确定其可用性并返回可用GPU的有序列表。
可用性是GPUtil的基础GPUtil是一个Python模块,用于使用nvidia-smi从NVIDA GPU获取GPU状态。
GPUtil查找计算机上的所有GPU,确定其可用性并返回可用GPU的有序列表。
可用性基于当前的内存消耗和每个GPU的负载。
该模块在编写时就考虑到了深度学习的GPU选择,但它不是特定于任务/库的,它可以应用于任何任务,在此情况下识别可用的GPU可能会很有用。
目录表
1