mindspore.device_context.gpu.device_count

查看源文件
mindspore.device_context.gpu.device_count()[源代码]

返回可用的GPU数量。

返回:

int,可用的GPU数量。

样例:

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