device_count

paddle.device.cuda. device_count ( )

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

返回

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

代码示例

COPY-FROM: paddle.device.cuda.device_count