CUDAPlace

class paddle. CUDAPlace

CUDAPlace is a descriptor of a device. It represents a GPU device allocated or to be allocated with Tensor. Each CUDAPlace has a dev_id to indicate the graphics card ID represented by the current CUDAPlace, staring from 0. The memory of CUDAPlace with different dev_id is not accessible. Numbering here refers to the logical ID of the visible graphics card, not the actual ID of the graphics card. You can set visible GPU devices by setting the CUDA_VISIBLE_DEVICES environment variable. When the program starts, visible GPU devices will be numbered from 0. If CUDA_VISIBLE_DEVICES is not set, all devices are visible by default, and the logical ID is the same as the actual ID.

Parameters

id (int) – GPU device ID.

Examples

>>> 
>>> import paddle
>>> place = paddle.CUDAPlace(0)

custom_device_id ( self: paddle.base.libpaddle.Place ) → int

custom_device_id¶

custom_device_type ( self: paddle.base.libpaddle.Place ) → str

custom_device_type¶

gpu_device_id ( self: paddle.base.libpaddle.Place ) → int

gpu_device_id¶

ipu_device_id ( self: paddle.base.libpaddle.Place ) → int

ipu_device_id¶

is_cpu_place ( self: paddle.base.libpaddle.Place ) → bool

is_cpu_place¶

is_cuda_pinned_place ( self: paddle.base.libpaddle.Place ) → bool

is_cuda_pinned_place¶

is_custom_place ( self: paddle.base.libpaddle.Place ) → bool

is_custom_place¶

is_gpu_place ( self: paddle.base.libpaddle.Place ) → bool

is_gpu_place¶

is_ipu_place ( self: paddle.base.libpaddle.Place ) → bool

is_ipu_place¶

is_xpu_pinned_place ( self: paddle.base.libpaddle.Place ) → bool

is_xpu_pinned_place¶

is_xpu_place ( self: paddle.base.libpaddle.Place ) → bool

is_xpu_place¶

set_place ( *args, **kwargs )

set_place¶

Overloaded function.

1. set_place(self: paddle.base.libpaddle.Place, arg0: paddle.base.libpaddle.Place) -> None

2. set_place(self: paddle.base.libpaddle.Place, arg0: phi::CPUPlace) -> None

3. set_place(self: paddle.base.libpaddle.Place, arg0: phi::XPUPlace) -> None

4. set_place(self: paddle.base.libpaddle.Place, arg0: phi::GPUPlace) -> None

5. set_place(self: paddle.base.libpaddle.Place, arg0: phi::GPUPinnedPlace) -> None

6. set_place(self: paddle.base.libpaddle.Place, arg0: phi::XPUPinnedPlace) -> None

7. set_place(self: paddle.base.libpaddle.Place, arg0: phi::IPUPlace) -> None

8. set_place(self: paddle.base.libpaddle.Place, arg0: phi::CustomPlace) -> None

xpu_device_id ( self: paddle.base.libpaddle.Place ) → int

xpu_device_id¶