device

class paddle.cuda. device ( device: Place | int | str | None = None ) [source]

Context-manager that changes the selected device.

Parameters

device (paddle.Place, int or str) – device index to select.

Examples

>>> import paddle

>>> print(paddle.device.get_device())  # gpu:0
>>> with paddle.device.device("cpu"):
...     print(paddle.device.get_device())  # cpu

>>> # paddle.cuda.device is an alias of paddle.device.device
>>> with paddle.cuda.device("cpu"):
...     print(paddle.device.get_device())  # cpu
>>> print(paddle.device.get_device())