get_device

paddle.get_device ( ) [源代码]

该功能返回当前程序运行的全局设备，返回的是一个类似于 cpu 或者 gpu:0 字符串，如果没有设置全局设备，当cuda可用的时候返回 gpu:0 ，当cuda不可用的时候返回 cpu 。

返回：返回当前程序运行的全局设备。

代码示例

  1. import paddle
  3. device = paddle.get_device()