is_compiled_with_cuda

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

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

返回

bool,支持 GPU 则为 True,否则为 False。

代码示例

import paddle
support_gpu = paddle.device.is_compiled_with_cuda()

使用本API的教程文档