modelbox.Device

函数 作用
get_device_id 获取当前设备编号, e.g. 0, 1, 2
get_type 获取当前设备类型, e.g. cpu, cuda, ascend
get_device_desc 获取当前设备描述

modelbox.Device.get_device_id

获取当前设备编号。

args:

return:

str 设备编号

modelbox.Device.get_type

获取当前设备类型。

args:

return:

str 设备的类型信息

modelbox.Device.get_device_desc

获取当前设备描述。

args:

return:

str 设备的描述信息

example:

    ...
    def process(self, data_ctx):
        device = get_bind_device()
        print(device.get_device_id())
        print(device.get_device_type())
        print(device.get_device_desc())
        ...

        return modelbox.Status()

result:

"0"
"cpu"
"this is a cpu device"

©2022 ModelBox Team all right reserved,powered by Gitbook文件修订时间: 2022-09-16 08:17:12

results matching ""

    No results matching ""