mindspore.device_context.gpu.device_count

mindspore.device_context.gpu.device_count()

Returns the number of GPUs available.

Returns

int, the number of GPUs available.

Examples

>>> import mindspore as ms
>>> ms.device_context.gpu.device.device_count()