PaddlePaddle

 paddle / is_compiled_with_cuda


is_compiled_with_cuda

paddle. is_compiled_with_cuda ( ) [源代码]

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

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

返回:是否支持GPU的bool值

示例代码

import paddle
support_gpu = paddle.is_compiled_with_cuda()

此页内容是否对您有帮助