is_compiled_with_custom_device

paddle.device. is_compiled_with_custom_device ( device_type ) [源代码]

检查 whl 包是否可以被用来在指定类型的自定义新硬件上运行模型

返回

bool,支持指定 device_type 则为 True,否则为 False。

代码示例

import paddle
support_npu = paddle.device.is_compiled_with_custom_device("npu")