mindspore_lite.GPUDeviceInfo
- class mindspore_lite.GPUDeviceInfo(device_id=0, enable_fp16=False)[源代码]
用于设置GPU设备信息的Helper类,继承自DeviceInfo基类。
参数:
device_id (int,可选) - 设备id。默认值:0。
enable_fp16 (bool,可选) - 启用以执行float16推理。默认值:False。
异常:
TypeError - device_id 不是int类型。
TypeError - enable_fp16 不是bool类型。
ValueError - device_id 小于0。
样例:
>>> import mindspore_lite as mslite >>> gpu_device_info = mslite.GPUDeviceInfo(device_id=1, enable_fp16=False) >>> print(gpu_device_info) device_type: DeviceType.kGPU, device_id: 1, enable_fp16: False. >>> cpu_device_info = mslite.CPUDeviceInfo(enable_fp16=False) >>> context = mslite.Context() >>> context.append_device_info(gpu_device_info) >>> context.append_device_info(cpu_device_info) >>> print(context) thread_num: 0, inter_op_parallel_num: 0, thread_affinity_mode: 0, thread_affinity_core_list: [], enable_parallel: False, device_list: 1, 0, .