device_count

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

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

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

代码示例

import paddle

paddle.device.cuda.device_count()