torch.cuda.graph_annotations.is_available

torch.cuda.graph_annotations.is_available() → bool [исходный код]

Возвращает, поддерживается ли запись аннотаций графа CUDA.

Для этого требуется устройство CUDA, пакет cuda-bindings и драйвер, поддерживающий cudaGraphNodeGetToolsId (CUDA >= 13.1 или эквивалентный пакет cuda-compat). Если возвращается False, mark_kernels() ничего не делает и аннотации не записываются.

При первом вызове может выполняться проверка драйвера CUDA; результат кэшируется.

Тип возвращаемого значения:

bool