mindspore.device_context.gpu.device_count

View Source On Gitee
mindspore.device_context.gpu.device_count()[source]

Return the number of GPUs available.

Returns

Bool, the number of GPUs available.

Examples

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