mindspore.hal.get_device_properties
- mindspore.hal.get_device_properties(device_id, device_target=None)[source]
Get specified device's properties.
Note
If device_target is not specified, get the device properties of the current backend set by context. For Ascend, backend must be initialized before calling this method, or total_memory and free_memory will be 0, and device_id will be ignored since this method only returns current device's properties.
- Parameters
- Returns
cudaDeviceProp for GPU.
cudaDeviceProp { name(str), major(int), minor(int), is_multi_gpu_board(int), is_integrated(int), multi_processor_count(int), total_memory(int), warp_size(int) }
AscendDeviceProperties for Ascend.
AscendDeviceProperties { name(str), total_memory(int), free_memory(int) }
None for CPU.
Examples
>>> import mindspore as ms >>> device_target = ms.context.get_context("device_target") >>> print(ms.hal.get_device_properties(0, device_target))