[ 组合替代实现 ]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())