paddle / set_device

set_device¶

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

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

参数：

device (str)- 此参数确定特定的运行设备，它可以是 cpu 、 gpu:x 或者是 xpu:x 。其中， x 是GPU 或者是 XPU 的编号。当 device 是 cpu 的时候，程序在CPU上运行，当device是 gpu:x 的时候，程序在GPU上运行。

返回：Place, 设置的Place。

代码示例

import paddle

paddle.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)

此页内容是否对您有帮助

PaddlePaddle

paddle

paddle.amp

paddle.callbacks

paddle.compat

paddle.distributed

paddle.distribution

paddle.fluid

paddle.io

paddle.jit

paddle.metric

paddle.nn

paddle.onnx

paddle.optimizer

paddle.regularizer

paddle.static

paddle.sysconfig

paddle.text

paddle.utils

paddle.vision

paddle / set_device

set_device¶