torch.nn.attention.is_cudnn_sdp_python_available

torch.nn.attention.is_cudnn_sdp_python_available()

Warning

This API is beta and subject to change.

Returns whether nvidia-cudnn-frontend’s cudnn.torch module can be found. It imports the cudnn package to look, but not cudnn.torch, and does not check the version: enable_cudnn_sdp_python() still raises if the installed package is older than it requires.

Return type:

bool