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