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")