torch.cuda.profiler.profile

torch.cuda.profiler.profile()

Enable profiling.

Context Manager to enabling profile collection by the active profiling tool from CUDA backend. .. rubric:: Example

>>> import torch
>>> model = torch.nn.Linear(20, 30).cuda()
>>> inputs = torch.randn(128, 20).cuda()
>>> with torch.cuda.profiler.profile() as prof:
...     model(inputs)