torch.cuda.is_current_stream_capturing
- torch.cuda.is_current_stream_capturing()[source]
Returns True if CUDA graph capture is underway on the current CUDA stream, False otherwise.
If a CUDA context does not exist on the current device, returns False without initializing the context.