current_device

paddle.cuda. current_device ( ) [源代码] 

返回当前设备的索引。

返回

int, 当前设备的索引。

代码示例

          >>> import paddle
>>> device_id = paddle.device.current_device() # this is equivalent to paddle.cuda.current_device()
>>> print(f"Current device index: {device_id}")

         