xpu_places¶

paddle.static. xpu_places ( device_ids=None ) [源代码] ¶

注解

多卡任务请先使用 FLAGS_selected_xpus 环境变量设置可见的 XPU 设备，下个版本将会修正 XPU_VISIBLE_DEVICES 环境变量无效的问题。

该接口根据 device_ids 创建一个或多个 paddle.XPUPlace 对象，并返回所创建的对象列表。

如果 device_ids 为 None，则首先检查 FLAGS_selected_xpus 标志。例如：FLAGS_selected_xpus=0,1,2，则返回的列表将为 [paddle.XPUPlace(0), paddle.XPUPlace(1), paddle.XPUPlace(2)]。如果未设置标志 FLAGS_selected_xpus，则根据 XPU_VISIBLE_DEVICES 环境变量，返回所有可见的 XPU places。

如果 device_ids 不是 None，它应该是使用的 XPU 设备 ID 的列表或元组。例如：device_id=[0,1,2]，返回的列表将是 [paddle.XPUPlace(0), paddle.XPUPlace(1), paddle.XPUPlace(2)]。

参数¶

device_ids (list(int)|tuple(int)，可选) - XPU 的设备 ID 列表或元组。默认值为 None。

返回¶

list[paddle.XPUPlace]，创建的 paddle.XPUPlace 列表。

代码示例¶

          >>> import paddle
>>> import paddle.static as static
>>> paddle.device.set_device('xpu')

>>> paddle.enable_static()
>>> xpu_places = static.xpu_places()

         