nvidia_xml.py 文件源码

def monitor(gpu_property_string, num_gpus=4):
    """"
    Parameters
    ----------
    gpu_property_string: temperature or utilization
    """
    data_dict = {'gpu {}'.format(gpu): [] for gpu in range(num_gpus)}
    data_dict['index'] = []
    data = ColumnDataSource(data=data_dict)
    p = gpu_plot(data, num_gpus=4, plot_width=600, plot_height=400, y_range=(0, 110))
    return p, create_running_func(data, _PROPERTY_DICT[gpu_property_string])