用于自动监控和占用NVIDIA GPU的Python软件包
GPU4U在计算机上定位所有GPU,确定其可用性并返回可用GPU的有序列表。可用性基于当前的内存消耗和每个GPU的负载。该软件包在编写时就牢记了用于深度学习的GPU选择,但它不是特定于任务/库的,它可以应用于任何任务,在其中识别可用的GPU可能会很有用。
要求
已安装最新NVIDIA驱动程序的NVIDIA GPU。 GPU4U使用程序nvidia-smi来获取所有NVIDIA GPU的GPU状态。在安装NVIDIA驱动程序时,应自动安装