xpu_places¶
注解
多卡任务请先使用 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
列表。
代码示例¶
# required: xpu
import paddle
import paddle.static as static
paddle.enable_static()
xpu_places = static.xpu_places()