import GPUtil

def getGpu():
    # 获取所有可用的GPU列表
    gpus = GPUtil.getGPUs()
    
    # print(gpus)
    # 没有找到GPU的情况
    if not gpus:
        return None

    # 找出剩余显存最多的GPU
    max_memory_gpu = max(gpus, key=lambda gpu: gpu.memoryFree)

    # 返回GPU的序号
    return max_memory_gpu.id

if __name__ == "__main__":
    # 使用该函数并打印结果
    gpu_id = getGpu()
    if gpu_id is not None:
        print(f"GPU with most free memory: GPU {gpu_id}")
    else:
        print("No GPUs found")