device_count

paddle.device.cuda. device_count ( ) [源代码]

返回值是 int,表示当前程序可用的 GPU 数量。

返回

返回一个整数,表示当前程序可用的 GPU 数量。

代码示例

import paddle

paddle.device.cuda.device_count()