29f42297a50a71acbdc469c4badd440d2ddfbf3a,finetune/config.py,,all_gpus,#,13

Before Change

    
    Get integer ids of all available GPUs
    
    local_device_protos = device_lib.list_local_devices()
    return [
        int(x.name.split(":")[-1]) for x in local_device_protos
        if x.device_type == "GPU"
    ]

GridSearchable = namedtuple("GridSearchable", "default iterator")

After Change

    
    sp = subprocess.Popen(["nvidia-smi", "-L"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    response = sp.communicate()[0]
    gpu_list = response.decode("utf-8").strip().split("\n")
    device_ids = []
    for i, gpu in enumerate(gpu_list):
        // May be worth logging GPU description

Instances