is_compiled_with_rocm¶

paddle.device. is_compiled_with_rocm ( ) [source]

Whether this whl package can be used to run the model on AMD or Hygon GPU(ROCm).

Returns

True if ROCm is currently available, otherwise False.

Return type

Bool

Examples

>>> import paddle
>>> support_gpu = paddle.device.is_compiled_with_rocm()