set_device

paddle.cuda. set_device ( device: DeviceLike ) → None [source]

Set the current device.

Parameters

device (DeviceLike) – The device to set as current. Can be paddle.CUDAPlace, paddle.CustomPlace, paddle.XPUPlace, int (device index), or str (device string).

Returns

None

Examples

>>> 
>>> import paddle
>>> # Set current device to GPU:0
>>> paddle.cuda.set_device(0)
>>> # Set current device to GPU:0
>>> paddle.cuda.set_device('gpu:0')
>>> # Set current device to a specific CUDAPlace
>>> place = paddle.CUDAPlace(0)
>>> paddle.cuda.set_device(place)