is_compiled_with_rocm¶

paddle. is_compiled_with_rocm ( ) [源代码] ¶

检查 whl 包是否可以被用来在AMD或海光GPU(ROCm)上运行模型

返回：bool，支持GPU(ROCm)则为True，否则为False。

示例代码

import paddle
support_gpu = paddle.is_compiled_with_rocm()