is_compiled_with_cuda

paddle.fluid.is_compiled_with_cuda()[源代码]

检查 whl 包是否可以被用来在GPU上运行模型

返回:支持gpu则为True,否则为False。

返回类型:out(boolean)

示例代码

import paddle.fluid as fluid
support_gpu = fluid.is_compiled_with_cuda()