set_device

paddle.device. set_device ( device ) [源代码]

Paddle 支持包括 CPU 和 GPU 在内的多种设备运行,设备可以通过字符串标识符表示,此功能可以指定 OP 运行的全局设备。

参数

  • device (str)- 此参数确定特定的运行设备,它可以是 cpugpuxpumlunpugpu:xxpu:xmlu:x 或者是 npu:x。其中,x 是 GPU、 XPU、 MLU 或者是 NPU 的编号。当 devicecpu 的时候,程序在 CPU 上运行,当 device 是 gpu:x 的时候,程序在 GPU 上运行,当 device 是 mlu:x 的时候,程序在 MLU 上运行,当 device 是 npu:x 的时候,程序在 NPU 上运行。

返回

Place,设置的 Place。

代码示例

>>> import paddle

>>> paddle.device.set_device("cpu")
>>> x1 = paddle.ones(name='x1', shape=[1, 2], dtype='int32')
>>> x2 = paddle.zeros(name='x2', shape=[1, 2], dtype='int32')
>>> data = paddle.stack([x1,x2], axis=1)