torch.cuda.graph_annotations.is_available

torch.cuda.graph_annotations.is_available() → bool

Return whether CUDA graph annotation recording is supported.

Requires a CUDA device, the cuda-bindings package, and a driver that supports cudaGraphNodeGetToolsId (CUDA >= 13.1 or an equivalent cuda-compat package). When this returns False, mark_kernels() is a silent no-op and no annotations are recorded.

The first call may probe the CUDA driver; the result is cached.

Return type:

bool