[ 组合替代实现 ]torch.backends.cudnn.is_available

torch.backends.cudnn.is_available

torch.backends.cudnn.is_available()

检测当前 cudnn 是否可用。

PaddlePaddle 目前无对应 API,可使用如下代码组合实现该 API。

转写示例

# PyTorch 写法
torch.backends.cudnn.is_available()

# Paddle 写法
bool(paddle.device.get_cudnn_version())