get_device
该功能返回当前程序运行的全局设备,返回的是一个类似于 cpu、 gpu:x、 xpu:x、 mlu:x 或者 npu:x 字符串,如果没有设置全局设备,当 cuda 可用的时候返回 gpu:0,当 cuda 不可用的时候返回 cpu 。
返回
返回当前程序运行的全局设备。
代码示例
>>> import paddle
>>> device = paddle.device.get_device()
>>> x_cpu = paddle.to_tensor([1, 2, 3], place=paddle.CPUPlace())
>>> id = paddle.get_device(x_cpu) # -1